Artificial Intelligence Specialist

We are looking for an experienced engineer who can help drive the direction of our ML platform at Jane Street, a global business with a unique trading environment that serves as a rapid-feedback platform for ML experimentation.

Our ML team is comprised of people with a shared love for software engineering and designing APIs and systems that are delightful to use. We rely on your in-depth knowledge of the ML ecosystem and understanding of various approaches to aid decision-making and apply the right tool for the problem at hand.

Your work will focus on enhancing research workflows to tighten our feedback cycles, and you will be able to understand the mechanics behind various modelling techniques and break down the mathematics behind them.

If you have a curious mind and a passion for solving interesting problems, we have a feeling you'll fit right in. We're looking for someone with experience building and maintaining training and inference infrastructure, a strong mathematical background, and a passion for keeping up with the state of the art.

Expertise in wrangling an ML framework, such as PyTorch, Jax, or TensorFlow, and fluency in English are also required. We're looking for someone with an inventive approach and the willingness to ask hard questions about whether we're taking the right approaches and using the right tools.

We're an equal opportunities employer and welcome applications from diverse backgrounds.