Jane Street
Machine Learning Research Engineer
New York, New York, United States · Full-time · Posted Nov 18, 2022
About the job
Build and maintain machine-learning infrastructure, research workflows, and tools that connect research ideas with production systems for Jane Street’s trading business.
What you'll do
- Enhance research workflows and APIs to accelerate experimentation and support varied use cases.
- Build and maintain training and inference infrastructure, taking machine-learning systems from concept to production.
- Collaborate with researchers to prototype ideas and advance model capabilities across the technology stack.
What they ask for
- Experience building and maintaining training and inference infrastructure, with an understanding of moving systems from concept to production.
- Strong mathematical background, including areas such as optimization theory, regularization, or linear algebra.
- Ability to create and maintain organized research codebases that produce robust, reproducible results.
- Expertise with an ML framework.
Nice to have
- Familiarity with PyTorch, Jax, TensorFlow, or other ML frameworks.
- Interest in academic research, new hardware, and the state of the art in machine learning.
Tools
PyTorch, Jax, TensorFlow
AI focus
Machine-learning research and engineering, including model development, training and inference infrastructure, research workflow improvements, and production deployment.
Model-written summary · the original posting is authoritative