Member of Technical Staff, Integration/RL Team (Research Engineer)

Extremely strong software engineering skills. Value test-driven development methods, clean code, and strive to reduce technical debts. Proficiency in Python and related ML frameworks such as JAX, Pytorch and/or XLA/MLIR. Experience using and debugging large-scale distributed training strategies. [Bonus] Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray). [Bonus] Hands-on experience with the post-training phase of model training. [Bonus] Experience in ML, LLM and RL academic research.

Design and write high-performing and scalable software for training models. Develop new tools to support and accelerate research and LLM training. Coordinate with other engineering teams and scientific teams. Craft and implement techniques to improve performance and speed up training cycles. Research, implement, and experiment with ideas on cluster and data infrastructure. Collaborate with other scientists, engineers, and teams.