Senior Machine Learning Engineer

Job Description:

We are a rapidly expanding AI startup that is transforming digital advertising with groundbreaking AI technology and innovative solutions. We are seeking an experienced Senior Machine Learning Engineer to build and deploy advanced ML systems that predict consumer behavior at scale.

As a Senior Machine Learning Engineer, you will design, train, and optimize production-grade ML models processing petabytes of consumer data and delivering billions of predictions daily. You will work with cutting-edge deep learning techniques and scalable ML infrastructure on AWS, ensuring high performance and reliability.

Key Responsibilities:

Design, develop, and train production-grade ML models using state-of-the-art algorithms and deep learning techniques for consumer behavior prediction

Implement distributed training workflows across multiple compute nodes

Optimize model architectures, training processes, and inference performance

Collaborate with data engineers for designing feature engineering pipelines

Conduct model experimentation, A/B testing, validation, and performance analysis

Configure distributed training parameters and resource allocation for large-scale training

Work with MLOps team to transition models from development to production

Document model architectures, training methodologies, and performance benchmarks

Stay current with ML research and drive innovation in model development

Qualifications:

Bachelor's/Master's/PhD in Computer Science, Machine Learning, AI, or related field

3+ years building and training production ML models at scale

Strong Python expertise with modern ML frameworks (PyTorch, TensorFlow, XGBoost)

Experience with distributed training frameworks and large-scale model training

Proficiency with AWS cloud platforms and GPU-based training

Strong understanding of supervised/unsupervised learning, deep learning, and transformer architectures

Experience with feature engineering and large-scale data processing

Knowledge of model optimization and performance tuning techniques

Strong experimentation, validation, and A/B testing skills

Excellent problem-solving and cross-functional collaboration abilities

Nice to have:

Prior experience in AdTech or MarTech industries

Knowledge of ML workflow orchestration tools

Familiarity with feature stores and data lake architectures

Model compression and optimization experience

Publications or open-source ML contributions

Startup environment experience

What we offer:

We offer a competitive salary, comprehensive benefits, and a dynamic work environment. If you are passionate about building revolutionary ML models that power AI-driven consumer insights, and want to be part of a rapidly growing startup transforming the advertising industry, we encourage you to apply for this exciting opportunity!