Machine Learning Engineer

Jane Street Jane Street · Quant · London, United Kingdom · Machine Learning

Machine Learning Engineer at Jane Street to drive the direction of their ML platform, focusing on building and maintaining training and inference infrastructure, enhancing research workflows, and applying ML to financial trading. Requires strong mathematical foundations and experience with ML frameworks.

What you'd actually do

  1. Experience building and maintaining training and inference infrastructure, with an understanding of what it takes to move from concept to production
  2. A strong mathematical background; Good candidates will be excited about things like optimisation theory, regularisation techniques, linear algebra and the like
  3. A passion for keeping up with the state of the art, whether that means diving into academic papers, experimenting with the latest hardware or reading the source of a new machine learning package
  4. A proven ability to create and maintain an organised research codebase that produces robust, reproducible results while maintaining ease of use
  5. Expertise wrangling an ML framework – we're fans of PyTorch, but we'd also love to learn what you know about Jax, TensorFlow or others

Skills

Required

  • Machine learning
  • mathematical foundations
  • ML ecosystem knowledge
  • neural networks
  • random forests
  • gradient-boosted trees
  • ensemble methods
  • optimisation theory
  • regularisation techniques
  • linear algebra
  • PyTorch
  • Jax
  • TensorFlow
  • Python

Nice to have

  • finance experience

What the JD emphasized

  • robust experience in machine learning
  • strong mathematical foundations
  • training and inference infrastructure
  • organised research codebase
  • robust, reproducible results

Other signals

  • ML platform development
  • enhancing research workflows
  • production ML systems