Senior Machine Learning Engineer, Data Mining
Core
Design massive multimodal Teacher models and distill them into hyper-efficient Student models to scour exabytes of sensor data for rare edge cases and model errors in autonomous driving.
Role type
Senior IC machine learning engineer (data mining & model distillation)
Builds
Omnitag, an ML-powered multimodal data mining framework for autonomous driving
Domain
Autonomous driving, multimodal sensor data, large-scale representation learning
Deliverable
production ML models
Required skills
model distillation, teacher-student training, reinforcement learning, Python, PyTorch/TensorFlow/JAX, CI/CD, cloud deployment (AWS/GCP/Azure), system design
Preferred skills
agentic systems, chain-of-thought prompting, autonomous driving/robotics, multimodal learning/sensor fusion, active learning, MLOps platforms
Technologies
PyTorch, TensorFlow, JAX, PPO, DQN, actor-critic, TF Serving, Triton, TorchServe
Responsibilities
Architect and train distilled models for multimodal sensor data; Build RL-based policy learning and reasoning systems; Optimize model deployment for real-time inference; Research and integrate agentic systems; Drive production reliability; Mentor and collaborate with ML scientists and data engineers
Seniority
Senior, hands-on IC