Machine Learning Research Engineer

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.

Responsibilities:

• 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.

Qualifications:

• 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++.

Benefits:

• 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.