Senior Machine Learning Engineer, Features
About The Position
In this role, you will work at the intersection of large-scale distributed systems, machine learning engineering, and platform architecture. You will own and evolve critical data capabilities within Cognitiv’s advertising ecosystem, with a primary focus on designing and delivering stable, highly scalable feature pipelines, as well as optimizing the performance and reliability of ad model training pipelines. This is a senior individual-contributor role for an engineer who thrives on solving complex platform problems, raising the technical bar, and building systems that serve many teams at scale. You will partner closely with engineers, product managers, data scientists, infrastructure teams, and downstream data consumers to deliver platforms that are resilient, extensible, and easy to adopt. This position will be located in San Mateo, CA with a hybrid work schedule of 3 days in office (Mon/Tue/Wed) and 2 days remote optional (Thursday/Friday).
Requirements
- 3-5+ years of hands-on Machine Learning Infrastructure / Data Platform experience supporting data-intensive platforms, including large-scale data pipelines, streaming systems, and storage layers.
- Proficiency in one or more core programming languages — Python, Java, or Scala — for building, maintaining, and scaling robust ML and data pipelines.
- Strong expertise in modern big data technologies such as Apache Spark, Apache Flink, Apache Kafka, and other distributed data processing frameworks.
- Excellent team communication, cross-functional collaboration, and problem-solving skills, with a track record of partnering effectively with modeling and engineering teams.
- Willing to work onsite Monday/Tuesday/Wednesday in San Mateo, CA.
Nice To Haves
- Domain expertise in AdTech.
- Hands-on experience with PyTorch for model architecture, training pipeline acceleration, or distributed training.
- Proficiency with cloud & infrastructure technologies, including AWS/GCP, as well as containerization and orchestration platforms like Kubernetes (K8s) and Docker.
- Competitive programming background, such as awards or achievements in OI (Olympiad in Informatics) or ACM/ICPC.
Responsibilities
- Architect, build, and maintain low-latency, high-throughput feature pipelines — batch, real-time streaming, and point-in-time correct historical features — to power our real-time bidding systems.
- Leverage LLMs and deep learning models to extract rich contextual and user-level embeddings into the core feature store/serving system, optimizing embedding generation, indexing, and online retrieval for sub-millisecond serving SLAs.
- Own and continuously enhance Cognitiv's ad model training pipelines, improving training speed, resource utilization, and throughput for large-scale deep learning models.
- Establish technical standards for monitoring, testing, and CI/CD across feature and training infrastructure to ensure robust system SLAs/SLOs.
- Translate complex signals into production-ready features that directly boost model performance (e.g., CTR/CVR prediction).
Benefits
- Medical, Dental and Vision plan for US employees & Extended Health Benefits for Canadian employees
- 12 weeks paid parental leave + 4 weeks WFH
- Unlimited PTO + Work-From-Anywhere August
- Career development with clear advancement paths
- Equity for all employees
- Hybrid work model & daily team lunch
- Health & wellness stipend + cell phone reimbursement
- 401(k) & RRSP with employer match
- Parking (CA, WA, Vancouver offices) & pre-tax commuter benefits
- Employee Assistance Program
- Comprehensive onboarding (Cognitiv University)
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
