Senior Machine Learning Research Engineer I/ II, Open-Endedness

Lila Sciences is seeking a Senior Machine Learning Research Engineer to join their Open-Endedness Team, led by Ken Stanley. This team focuses on developing systems that transcend traditional problem-solving by fostering continuous, transformative creativity in scientific discovery. Lila Sciences is pioneering the integration of AI into the scientific method to address humanity's most pressing challenges.

In this role, you will design and maintain large-scale training systems, optimize performance for large models, and integrate cutting-edge techniques to enhance efficiency and throughput. Your work will involve building distributed training infrastructure for large language models (LLMs) and multi-modal models, orchestrating frontier and open-source LLMs, and developing scalable pipelines for data preprocessing and experiment orchestration.

The ideal candidate will have proven experience with distributed machine learning training frameworks such as Megatron-LM, TorchTitan, DeepSpeed, or Ray. Strong software engineering skills in Python are essential, with C++ kernel contributions considered a plus. A solid understanding of large-scale model training techniques and experience with cloud or high-performance computing environments are also required.

Lila Sciences offers competitive compensation, including bonus potential and generous early equity. The expected base salary range for this position is $148,000 to $240,000 USD. Additional benefits include comprehensive medical, dental, and vision coverage, flexible time off, paid parental leave, and a company-subsidized lunch program.

Joining Lila Sciences means becoming part of a mission-driven team dedicated to accelerating scientific discovery through AI. The company fosters a collaborative and inclusive environment, encouraging innovative thinking and providing opportunities for professional growth in the rapidly evolving field of AI-driven science.