Research Engineer
About the Role
Work alongside an AI research team to turn experimental code into production-ready systems. You will build and improve shared post-training infrastructure that supports research data generation, processing, and evaluation.
What You'll Do
Turn research prototypes into tested, reusable data generation, training, and evaluation pipelines.
Build distributed experiment support, including data loading, checkpointing, and collection of agent interactions.
Profile workloads and improve GPU utilization, memory efficiency, and data throughput.
Develop tests, experiment tracking, and debugging tools while preserving the integrity of research results.
Collaborate with researchers to translate ideas into maintainable software.
What We're Looking For
Experience building ML training, inference, or data pipelines in Python.
Strong Python skills and hands-on experience with PyTorch, JAX, or comparable machine learning frameworks.
Understanding of machine learning experiments, including how data, numerical precision, and implementation choices affect result
