ML Platform Engineer: From Research to Production

TheDataJob; New York (NY); On-site; USD 150,000 - 210,000; Full time

Job summary

Jane Street is seeking an engineer with strong machine learning foundations to advance our ML platform and research workflows. You will contribute to API design and systems that enable rapid experimentation in a finance-driven environment.

We value a deep mathematical background and experience building training and inference infrastructure, with familiarity across PyTorch, Jax, or TensorFlow. A curious mindset and passion for solving hard problems are highly regarded.

Qualifications

• Experience building and maintaining training and inference infrastructure.
• Strong mathematical background including optimization theory and linear algebra.
• Ability to design robust research codebases that produce reproducible results.
• Familiarity with ML frameworks such as PyTorch, Jax or TensorFlow.

Responsibilities

• Help drive the direction of the ML platform and its APIs.
• Enhance research workflows to tighten feedback cycles.
• Contribute to system design for scalable ML experiments.

Skills

Mathematics

Optimization theory

Linear algebra

Research/codebase organization

Curiosity/problem solving

Tools

PyTorch

Jax

TensorFlow

Job description

Jane Street is seeking an engineer with strong machine learning foundations to advance our ML platform and research workflows. You will contribute to API design and systems that enable rapid experimentation in a finance-driven environment.

We value a deep mathematical background and experience building training and inference infrastructure, with familiarity across PyTorch, Jax, or TensorFlow. A curious mindset and passion for solving hard problems are highly regarded.