ML Research Engineer, Performance Optimization

About the job

Your Impact at LILA Lila is building a platform where AI and automation co-evolve to accelerate scientific discovery Our AI Research team is looking to hire a ML Research Engineer to maximize iteration speed for mid- and post-training runs on 1T+ parameter LLMs using the latest NVIDIA chips. This involves not only maximizing GPU utilization on an individual training run but also making these optimizations maintainable, reusable, and adaptable so that the rest of the research team always knows how to confidently use the latest optimized stack. What You'll Be Building - GPU compute performance: profiling, kernel development, long-context handling, memory footprint optimization - GPU cluster…