Integral Ad Science (IAS) is looking for a Staff Machine Learning Engineer to oversee a sophisticated suite of data science systems making large scale business predictions in advertising inventory across various platforms. The role aims to push the boundaries of machine learning applications and deliver best-in-class solutions for clients, driving innovation and contributing to core products.
Requirements
- Experience working with machine learning frameworks such as TensorFlow, Caffe2, PyTorch, Spark ML, scikit-learn, or related frameworks
- Experience working with machine learning, ranking infrastructures, and system design
- Strong understanding of machine learning approaches and algorithms
- Strong programming skills in Python or Go
- Understanding of data structures, software design principles, and algorithms
Responsibilities
- Experience building large-scale deep learning infrastructure or platforms for distributed model training
- Experience with large-scale AI training infra components, such as accelerators, network fabrics, CUDA, NCCL, RDMA
- Experience building large-scale distributed systems with tools such as Kubernetes, Kafka, Prometheus, etc.
- Experience with deep learning frameworks, such as PyTorch, or JAX
- Drive system design for our AI/ML-based services.
- Design, develop and support our CI/CD pipeline for AI/ML-based services.
Other
- PhD/Master’s degree in technical field such as computer science, mathematics, statistics or equivalent years of experience
- 7+ years machine learning experience in industry
- Able to prioritize duties and work well on your own
- Ability to work with both internal and external partners
- Skilled at solving open ambiguous problems
- Strong collaboration and mentorship skills
