Skip to contentMachine Learning Performance Engineer
| Company | Optiver |
|---|
| Location | New York, NY, USA |
|---|
| Salary | $200000 – $200000 |
|---|
| Type | Full-Time |
|---|
| Degrees | |
|---|
| Experience Level | Mid Level, Senior |
|---|
Requirements
- Expertise in internals of deep-learning frameworks like PyTorch, JAX, TensorFlow, etc.
- Deep understanding of computer architecture
- Experience in programming in C++ and Python
Responsibilities
- Build scalable and robust training and inference pipelines for deep learning
- Dive into internals of open-source deep learning frameworks and enhance their functionality
- Identify and eliminate performance bottlenecks
- Collaborate closely with researchers and other engineers
- Develop an in-depth understanding of trading systems
Preferred Qualifications
- Experience with JAX ecosystem (XLA, Flax, etc.)
- Experience in programming for GPUs or other accelerators (CUDA, Triton, Pallas, etc.)
- Linux system programming experience
- Experience with large-scale distributed training
- Contributions to open-source projects related to data science and machine learning
