Jane Street Logo

Jane Street

Machine Learning Research Engineer

Reposted 2 Months Ago
In-Office
London, Greater London, England, GBR
Mid level
In-Office
London, Greater London, England, GBR
Mid level
Machine Learning Engineer needed to enhance ML platform, build infrastructure, and utilize ML frameworks to solve problems in a financial context.
The summary above was generated by AI

We are looking for a Research Engineer with robust experience in machine learning and strong mathematical foundations to join our growing ML team and to help drive the direction of our ML platform.

Machine learning is a critical pillar of Jane Street's global business. Our ever-evolving trading environment serves as a unique, rapid-feedback platform for ML experimentation, allowing us to incorporate new ideas with relatively little friction. Our ML team is full of people with a shared love for the craft of software engineering, and for designing APIs and systems that are delightful to use. 

We’ll rely on your in-depth knowledge of the ML ecosystem and understanding of varying approaches - whether it’s neural networks, random forests, gradient-boosted trees or sophisticated ensemble methods - to aid decision-making so we apply the right tool for the problem at hand. Your work will also focus on enhancing research workflows to tighten our feedback cycles. Successful ML Research Engineers will be able to understand the mechanics behind various modelling techniques, while also being able to break down the mathematics behind them. We’re looking for people who can be a true bridge between research and engineering, and can help us push forward the frontier of our model capabilities at all points of the stack. This could involve enabling a new scale in our technology stack, sitting on a team of Researchers to rapidly prototype a new research idea, or building workflow APIs that support a wide range of use cases.

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. While there isn’t a fixed list of qualifications we’re looking for, 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, with an understanding of what it takes to move from concept to production
  • A strong mathematical background; Good candidates will be excited about things like optimisation theory, regularisation techniques, linear algebra and the like
  • 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
  • A proven ability to create and maintain an organised research codebase that produces robust, reproducible results while maintaining ease of use
  • 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
  • An inventive approach and the willingness to ask hard questions about whether we're taking the right approaches and using the right tools
  • Fluency in English required

If you're a recruiting agency and want to partner with us, please reach out to [email protected].

Jane Street London, England Office

2 & A Half, Devonshire Square, London, United Kingdom, EC2M 4UJ

Similar Jobs

14 Days Ago
In-Office
London, Greater London, England, GBR
Entry level
Entry level
Artificial Intelligence • Machine Learning • Biotech • Pharmaceutical
Lead machine learning research for molecular foundation models, including architecture design, experimentation, structure generation, spectral retrieval, mass spectrum simulation, and molecular property prediction. Develop molecular and spectral representations, analyze model behavior, establish reproducible evaluation pipelines, and collaborate with computational chemists on leakage-safe benchmarks. Translate research findings into modeling priorities and R&D direction while communicating results to technical and non-technical stakeholders.
Top Skills: Computational Mass SpectrometryContainerized PipelinesFoundation ModelsGraph Neural NetworksInchikeyMolecular FingerprintsPythonSmilesTransformers
2 Months Ago
In-Office
London, Greater London, England, GBR
Senior level
Senior level
Healthtech • Machine Learning • Biotech
Design, implement and optimise large neural network models and scalable training pipelines. Build distributed multi-GPU/multi-node training, profile and optimise GPU workloads, maintain core ML infrastructure, and productionise research models for downstream biological discovery workflows.
Top Skills: Cloud-Based Ml InfrastructureContainerized Environments (Docker)CudaDistributed Training FrameworksFlashattentionGpuMulti-GpuMulti-NodePythonPyTorchTriton
3 Months Ago
Hybrid
London, Greater London, England, GBR
Mid level
Mid level
Artificial Intelligence • Machine Learning • Software • Nanotechnology
Implement and scale cutting-edge ML research into production-ready systems: build distributed training and inference on GPU clusters, optimize model performance, create evaluation and experiment-tracking tooling, architect multimodal scientific data pipelines, and collaborate with researchers and materials scientists to deploy models in closed-loop hardware environments.
Top Skills: Diffusion ModelsDistributed TrainingDockerGCPGnnsGpu ClustersJaxKubernetesMulti-GpuMulti-NodePythonPyTorchTransformers

What you need to know about the London Tech Scene

London isn't just a hub for established businesses; it's also a nursery for innovation. Boasting one of the most recognized fintech ecosystems in Europe, attracting billions in investments each year, London's success has made it a go-to destination for startups looking to make their mark. Top U.K. companies like Hoptin, Moneybox and Marshmallow have already made the city their base — yet fintech is just the beginning. From healthtech to renewable energy to cybersecurity and beyond, the city's startups are breaking new ground across a range of industries.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account