AI engineer - Distributed Systems

Responsibilities

• Work closely with the rest of the research team on experiment tracking and tooling to ensure large-scale training runs can be logged and analyzed with low overhead

• Automate & evolve the handling of python environments using tools such as docker and uv, as well as handling compilation of custom packages to ensure experiments and training runs can be reproduced 

• Set up & maintain CI/CD pipelines to automatically test large codebases, optimizing for test coverage

• Work on documentation & type correctness

• Define test suites to automatically test cluster stability & performance for distributed ML workloads

• Debug and resolve systems issues, ensuring that they are triaged & handled in a timely manner

Must have experience

• Excellent software engineering skills, particularly with experience in maintaining & working on typed & tested Pytorch code bases

• Experience with PyTorch

• Experience with Slurm

• Experience with Github CI/CD

• Experience with Docker

At Luma AI, we believe that multimodality is critical for intelligence. To go beyond language models and build more aware, capable and useful systems, the next step function change will come from vision. So, we are working on training and scaling up multimodal foundation models for systems that can see and understand, show and explain, and eventually interact with our world to effect change.

We will deploy these systems to make a new kind of intelligent creative partner that can imagine with us. Free and away from the pressure of being creative. It's for all of us whose imaginations have been constrained, who've had to channel vivid dreams through broken words, hoping others will see what we see in our mind's eye. A partner that can help us show — not just tell.

Dream Machine is an early step to building that. Try it here

Why you should join us:

• Luma is bringing together the best team in the world to achieve our goal, from researchers to engineers and designers to growth operators

• Luma is not just a lab - we are deeply product focused and our vision merging AI models and delightful products is unique in the industry

• We build. We ship. Our early products have been wildly successful