Senior Machine Learning Engineer, Cosmos Engineering

What you'll do

• Develop Cosmos-powered AI solutions for intelligent video analytics, perception, and physical AI use cases.
• Identify model capability gaps and recommend improvements such as synthetic data generation, tuning methods, and architecture changes.
• Partner with the Cosmos team and other functions on platform features, roadmap alignment, and open-sourcing team solutions and research artifacts.

What they ask for

• MSc or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
• 8+ years of experience in applied machine learning or AI research.
• Deep learning expertise, including hands-on experience with diffusion models and generative architectures.
• Experience with large-scale foundation models, including pre-training or refining LLMs, VLMs, or WFMs, and training, fine-tuning, and evaluation workflows.

Nice to have

• Experience with simulation environments such as Isaac Sim or similar platforms.
• Experience with model compression, quantization, real-time inference optimization, physical AI deployments, distributed AI systems, relevant research or open-source work, or CUDA, Triton, or GPU kernel development.

Pay

184,000 USD - 287,500 USD (base salary, Level 4); 224,000 USD - 356,500 USD (base salary, Level 5)

Tools

Cosmos, Isaac Sim, CUDA, Triton

AI focus

Develop production AI solutions using world foundation models for perception and physical AI; identify model gaps and improve capabilities through synthetic data, tuning, and architecture work.