Senior ML Systems Engineer, Frameworks & Tooling

Nice to Have

• Experience with training LLMs or other large transformer architectures.
• Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.).
• Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches).
• Experience with data pipeline optimization, sharded datasets, or caching strategies.
• Background in performance engineering, profiling, or low-level systems.

Bonus: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

If any of the above doesn’t line up exactly with your experience, we still encourage you to apply

We strive to create an inclusive work environment for all; we welcome applicants from all backgrounds and are committed to providing equal opportunities. Should you require any accommodations during the recruitment process, please submit an Accommodations Request Form, and we will work together to meet your needs.

We may use AI-enabled tools to screen and assess applicants against the criteria for this position. This helps our recruiters identify potentially qualified candidates, but it doesn't limit the applications our recruiters may review or consider.