Machine Learning Engineer

Jane Street Jane Street · Quant · Hong Kong · Machine Learning

Jane Street is seeking a Machine Learning Engineer to join their ML team and drive the direction of their ML platform. The role involves building and maintaining training and inference infrastructure, enhancing research workflows, and applying ML to financial trading. Requires a strong mathematical background and expertise in 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
  • Optimization theory
  • Regularization techniques
  • Linear algebra
  • PyTorch
  • Jax
  • TensorFlow
  • Software engineering
  • API design
  • System design
  • Research codebase management
  • Reproducible results

Nice to have

  • Curious mind
  • Passion for solving interesting problems

What the JD emphasized

  • building and maintaining training and inference infrastructure
  • move from concept to production
  • organised research codebase that produces robust, reproducible results

Other signals

  • ML platform development
  • Enhancing research workflows
  • Moving from concept to production