Senior Machine Learning Engineer - Generative Models

Job summary

Responsibilities

We are looking for senior machine learning engineers to advance the generative modeling technology behind Neural Simulation, our state-of-the-art product for turning real-world driving data into high-fidelity, photorealistic simulation environments.

• Develop and advance diffusion and video generation models that power our Neural Simulation product, bringing the latest research into production to improve realism…
• Push the limits of generative simulation for driving scenes, including:
• Controllable generation conditioned on scene layout, camera pose, actors, and trajectories
• Temporally consistent, multi-camera video generation
• Augmenting real-world logs with new scenarios, actors, and conditions such as weather and lighting

What they're looking for

• 5+ years of experience developing and shipping ML or computer vision systems
• A minimum of a Bachelor's degree in computer science, physics, robotics, or equivalent
• Strong hands-on experience with diffusion models and video generation (e.g., latent and video diffusion models, diffusion transformers)
• A solid foundation in generative modeling and deep learning, including training and fine-tuning large models
• Proficiency in Python and PyTorch

Tech stack

• Transformers
• Python
• PyTorch
• C++
• CUDA