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 transformer architectures and variants of attention mechanisms.
• Hands-on experience with large scale training (e.g., ZeRO, DDP, FSDP, TP, CP) and data processing tools (e.g., Ray, Spark).
• Production-quality software engineering skills in Python.
• MS, PhD, or equivalent experience in Computer Science, Machine Learning, Applied Math, Physics, or related field.
• 12+ years of relevant software development experience.

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

• Educational background with MS, PhD, or equivalent in relevant fields.

Benefits & Perks

• Base salary range of $224,000 - $356,500 USD, determined by location, experience, and similar positions.
• Eligibility for equity and benefits.