Senior Generative AI Research Engineer

About the Role

Key Responsibilities

• Design, post-train, and optimize foundation models (e.g., LLMs, diffusion video models, VLMs, VLAs) for real world applications.
• Contribute to highly-collaborative development on large-scale training infrastructure, high-efficiency inference pipelines, and scalable data pipelines.
• Work with teams in research, software, and product to bring world models from idea to deployment.
• Collaborate on open-source and internal projects, author technical papers or patents, and mentor junior engineers.
• Prototype and iterate rapidly on experiments across cutting-edge AI domains, including agentic systems, reinforcement learning, reasoning, and video generation.
• Design and implement model distillation algorithms for size reduction and diffusion step optimization.
• Profile and benchmark training and inference pipelines to achieve production-ready performance requirements.

Requirements

• Minimum 8 years industry or 5+ years research/postdoc experience building and deploying generative AI systems.
• Proficiency in PyTorch, JAX, or other deep learning frameworks.
• Expertise in one or more of: LLMs, coding agents, diffusion models, autoregressive models, VAE/GAN architectures, retrieval-augmented generation, neural rendering, or multi-agent systems.
• Experience with large scale training (e.g., ZeRO, DDP, FSDP, TP, CP) and data processing (e.g., Ray, Spark).
• Production-quality software engineering skills, with all work in Python.
• MS or PhD or equivalent experience in Computer Science, Machine Learning, Applied Math, Physics, or a related field.

Nice to Have

• Familiarity with high-performance computing and GPU acceleration.
• Contributions to influential open-source libraries or conference publications (NeurIPS, ICML, CVPR, ICLR).
• Experience working with multimodal data (e.g., vision-language, VLA, audio).
• Prior work with NVIDIA GPU-based compute clusters or simulation environments.

Qualifications

• MS or PhD or equivalent experience in relevant fields.

Benefits & Perks

• Base salary range from $224,000 to $356,500 for Level 5, and from $272,000 to $425,500 for Level 6, depending on location and experience.
• Eligibility for equity and benefits.
• Application deadline: July 29, 2025.