Machine Learning Research Engineer

Salary $150,000 - $250,000
Employment type Permanent

Job description

Job Title: Machine Learning Research Engineer

Location: Chicago, Illinois

Job Description: We are seeking a Machine Learning Research Engineer to join our team, focusing on a pivotal AI initiative. This role offers the opportunity to significantly impact Machine Learning infrastructure, training, and inference challenges to advance our futures trading strategies.


  • Conduct innovative research on deep learning for price forecasting.
  • Build scalable and robust training and inference pipelines for deep learning.
  • Enhance the functionality of open-source deep learning frameworks.
  • Collaborate closely with researchers and other engineers.
  • Develop an in-depth understanding of trading systems.


  • PhD or equivalent industry experience in a field related to machine learning.
  • Expertise in building deep-learning models in PyTorch, JAX, or TensorFlow.
  • Experience programming in Python.
  • Experience in computationally intensive research on very large data sets.

Preferred Qualifications:

  • Experience with the JAX ecosystem (XLA, Flax, etc.).
  • Experience in programming for GPUs or other accelerators (CUDA, Triton, Pallas, etc.).
  • Contributions to open-source projects related to data science and machine learning.
  • Strong publication record at conferences like NeurIPS, ICML, etc.
  • Expertise in the internals of deep-learning frameworks like PyTorch, JAX, TensorFlow, etc.
  • Experience with large-scale distributed training.
  • Experience programming in C++.


  • Work alongside best-in-class professionals from over 40 different countries.
  • Performance-based bonus structure that is unmatched in the industry, fostering a truly collaborative environment.
  • Ownership over initiatives that directly solve business problems.