Research Engineer, Machine Learning

What You Will Do

• Accelerate researchers by taking on the heavy parts of large-scale ML pipelines and building robust tools.

• Interface cutting-edge research with production: integrate checkpoints, streamline evaluation, and expose APIs.

• Conduct experiments on the latest deep-learning techniques (sparsified 70 B + runs, distributed training on thousands of GPUs).

• Design, implement and benchmark ML algorithms; write clear, efficient code in Python.

• Deliver prototypes that become production-grade components for Le Chat and our enterprise API.

• Master’s or PhD in Computer Science (or equivalent proven track record).

• 4 + years working on large-scale ML codebases.

• Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed / FSDP / SLURM / K8s).

• Experience in deep learning, NLP or LLMs; bonus for CUDA or data-pipeline chops.

• Strong software-design instincts: testing, code review, CI/CD.

• Self-starter, low-ego, collaborative.