Senior AI Research Engineer developing and deploying next - generation robot foundation models and end - to - end policies. Researching and implementing methods for humanoid robots to perform real - world tasks in a hybrid work environment.
About the role
- Senior Engineer in generative AI focused on foundation models and AI systems design at NVIDIA. Collaborating on large-scale training and contributing to open-source projects.
Responsibilities
- Design and post-train foundation models (LLMs, VLMs, VLAs and DiTs)
- Contribute to large-scale training infrastructure and high-efficiency inference pipelines
- Collaborate on open-source and internal projects, author technical papers or patents
- Prototype and iterate rapidly on experiments across AI domains
- Design and implement model distillation algorithms for size reduction and diffusion step optimization
- Profile and benchmark training and inference pipelines
Requirements
- Minimum 8 years industry or 5+ years research/postdoc in generative AI systems
- Proficiency in PyTorch, JAX, or other deep learning frameworks
- Expertise in LLMs, coding agents, diffusion models, autoregressive models, VAE/GAN architectures
- Familiarity with all variants of attention mechanisms
- Hands-on experience with large scale training (e.g., ZeRO, DDP, FSDP)
- Experience with data processing (e.g., Ray, Spark)
- MS or PhD or equivalent in Computer Science, Machine Learning, Applied Math, Physics, or related field
- 12+ years of relevant software development experience
Benefits
- Eligible for equity and benefits
Job title
Job type
Experience level
Salary
Degree requirement
Tech skills
Location requirements
Machine Learning Engineer at Salesforce AI Research working closely with teams to design agentic AI systems. Innovating at the frontier of AI with transformative solutions for Salesforce customers.
Principal Data Scientist leading design, development, and deployment of AI solutions for PEMCO's insurance portfolio. Working closely with teams to drive measurable impact through data science.
Cloud & AI Researcher joining PointFive to optimize cloud costs through research and innovative solutions. Collaborating with product teams to enhance platform features and publish technical content.
AI Data Scientist driving research and development for Pearson’s learner models and next - gen AI - driven learning products. Collaborating with teams to implement advanced AI strategies.
AI Research Lead defining research agenda for 1mind’s multimodal AI models. Leading exploratory research and building a world - class applied research team in a high - impact role.
AI Research Resident exploring foundational AI systems from first principles at Maincode, an AI research lab building systems that move humans forward.
AI Research Engineer developing sophisticated workflows leveraging LLM models at Cisco. Collaborating with teams to ensure security and scalability of AI solutions.
AI Scientist at Preply applying deep learning and NLP for personalized learning solutions. Collaborating with cross - functional teams to translate AI research into impactful educational tools.
Senior Applied AI Scientist focusing on AI Learning at Preply, leveraging deep learning for personalized education solutions. Collaborating with teams to enhance learning experiences globally.