Senior Machine Learning Engineer, Cosmos Engineering
$184k - $357k • Remote • US, CA, Santa Clara
Posted 1d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
Fully remote
Categories
Machine Learning Engineer
About the job
NVIDIA's Metropolis team is advancing physical AI by building intelligent video analytics and perception solutions for smart cities, industrial automation, and autonomous systems. We are seeking a Senior ML Engineer to develop and deliver next-generation Metropolis solutions powered by Cosmos, NVIDIA's world foundation model platform. This role involves constructing production-quality AI capabilities for Metropolis by demonstrating Cosmos to address real-world perception and physical AI challenges. You will also collaborate with the Cosmos team to enhance the platform's features in line with the Metropolis product roadmap. If you are passionate about pushing generative AI, simulation, and large-scale model development forward and want your work to impact real-world systems, we encourage you to apply.
Responsibilities
- Build and deliver innovative Metropolis AI solutions using Cosmos world foundation models to address real-world requirements in intelligent video analytics, perception, and physical AI.
- Identify areas where Cosmos models underperform or lack capability and propose new solutions, including synthetic data generation, tuning methods, and architecture improvements, using Metropolis use cases as a reference.
- Collaborate with the Cosmos team to identify and prioritize new platform features based on the Metropolis product roadmap, contributing to deliverables for the wider ecosystem.
- Lead the open-sourcing of solutions and research artifacts developed by the team.
- Stay current with the latest advances in foundation models, generative architectures, and training methodologies, and integrate relevant insights into team practices.
- Partner cross-functionally with Product, Program, Engineering, and Data Procurement teams to ensure alignment and facilitate execution.
Requirements
- MSc or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
- 8+ years of proven experience in applied machine learning or AI research.
- Deep expertise in deep learning fundamentals, with hands-on experience in diffusion models and generative architectures.
- Experience in pre-training or refining large language models (LLMs), vision-language models (VLMs), or world foundation models (WFMs).
- Experience working with large-scale foundation models, including training workflows, fine-tuning techniques, and evaluation approaches.
- Experience working with simulation environments like Isaac Sim or similar platforms.
- Consistent track record of leading projects end-to-end and delivering results with clarity and accountability.
- Ability to manage and complete tasks across multiple parallel workstreams in a fast-paced, evolving environment.
- End-to-end understanding of the ML development and deployment lifecycle, with the ability to quickly adopt and leverage modern AI development tools and workflows.
Benefits
- Competitive salary package
- Equity
- Benefits