Lead Machine Learning Engineer developing training systems to optimize multimodal robotic data processing. Collaborating with teams to enhance autonomy models and improve training efficiencies.
Responsibilities
Design and maintain training systems that can process and learn from petabyte-scale multimodal datasets (e.g., video and point cloud data). This includes ensuring data is efficiently loaded, distributed, and processed across large GPU clusters.
Identify and resolve bottlenecks in the training pipeline, including data loading, preprocessing, model computation, and inter-node communication, to maximize GPU utilization and reduce training time.
Work with the ML team to develop and refine neural network architectures suitable for autonomy tasks, particularly those handling high-dimensional and sequential sensor data.
Create and adjust loss functions and training strategies that help the model learn effectively from complex multimodal inputs and improve autonomy performance.
Configure, monitor, and maintain large-scale distributed training jobs across multiple machines and GPUs, ensuring stability, fault tolerance, and efficient resource usage.
Implement scalable systems to preprocess, transform, and augment large robotics datasets so that they are suitable for model training.
Work closely with ML scientists and other engineers to integrate new models, experiments, and training approaches into the production training pipeline.
Analyze training metrics, model outputs, and experiment logs to assess model performance and guide improvements in architecture, data usage, or training strategies.
Develop tools and workflows that allow teams to run experiments, track results, and iterate quickly on new model ideas or training approaches.
Requirements
Master’s or PhD in Computer Science, Robotics, Electrical Engineering, Machine Learning, or a closely related technical discipline.
Minimum of 5 years of professional experience developing, training, and deploying machine learning models in production environments.
Hands-on experience training machine learning models across multiple GPUs or compute nodes, including familiarity with distributed training frameworks and large dataset handling.
Strong programming skills in Python for implementing machine learning models, data pipelines, and training workflows.
Solid knowledge of core concepts such as neural networks, optimization algorithms, loss functions, model evaluation, and training methodologies.
Machine Learning Engineer developing consumer - facing AI features powered by LLMs at Quora. Collaborating with engineers and optimizing production AI systems for enhanced user engagement.
Staff Machine Learning Engineer on Merchant team at Pinterest. Leading ML initiatives to enhance merchant quality and improve user shopping experiences.
Machine Learning Engineer at Sumsub building ML solutions combating fraud and identity theft. Work with global teams in a fully remote setting allowing product autonomy.
Associate Deep Learning Engineer implementing deep learning and computer vision algorithms at NBCUniversal. Collaborating with cofounders to meet technical milestones and deploying code on geospatial datasets.
Staff Product Manager leading Vegetation Intelligence at Overstory, harnessing machine learning to enhance utility vegetation management. Overseeing product strategy and development for critical vegetation insights.
Machine Learning Engineering Intern developing innovative AI solutions for contact centers at Cresta. Collaborating with top researchers and engineers in AI to unlock customer experience potential.
MLOps/LLMOps Architect designing AI solutions for a next - gen AI platform in Canada. Collaborating with teams on architecture, deployment, and governance in AI environments.
Staff Deep Learning Engineer at NBCUniversal implementing core algorithms for 2D and 3D media datasets. Collaborating with cofounders and deploying large - scale algorithms for high - fidelity content.
Deep Learning Engineer at NBCUniversal implementing algorithms in computer vision and graphics. Transforming high dimensional data into high - fidelity content using advanced techniques.
Staff MLOps Engineer responsible for building infrastructure for AI/ML at NBCUniversal. Managing large media datasets, deploying models, and automating data pipelines.