Machine Learning Performance Engineer — Global Tech-Driven Trading Firm – New York ( full-time in office role ) – $300,000-$600,000 ( package ) depending on experience + Great benefits 

A leading global market maker, operating across more than 70 exchanges, is seeking a Machine Learning Performance Engineer to contribute to a pivotal AI initiative. This engineer will help build and optimize large-scale ML infrastructure that supports sophisticated futures trading strategies. The firm has continuously improved financial markets through technology-driven solutions, providing liquidity, accurate pricing, and transparency for participants worldwide. This role offered an opportunity to work at the intersection of high-performance computing, machine learning, and cutting-edge trading systems.

What the role involves

• Designing and building scalable, robust training and inference pipelines for deep learning models.
• Working deep within open-source deep learning frameworks (such as PyTorch, JAX, TensorFlow) to enhance their functionality and performance.
• Profiling complex systems and removing performance bottlenecks at software and hardware levels.
• Optimizing computing workloads across CPUs, GPUs, and specialized accelerators.
• Collaborating closely with researchers and trading engineers to deliver production-ready solutions.
• Gaining a comprehensive understanding of trading systems and contributing to continuous optimization.

What the ideal candidate brings

• Expertise in the internals of leading deep learning frameworks (PyTorch, JAX, TensorFlow).
• A deep understanding of computer architecture and systems-level optimization.
• Strong proficiency in C++ and Python for performance-critical applications.
• A track record of identifying and resolving performance bottlenecks at scale.

Preferred additional experience included:

• Familiarity with the JAX ecosystem (XLA, Flax).
• GPU programming expertise (CUDA, Triton, Pallas).
• Linux system programming experience.
• Experience with large-scale distributed training frameworks.
• Contributions to open-source machine learning or data science projects.

What the successful candidate receives

• The chance to work in a collaborative environment surrounded by best-in-class professionals from over 40 countries.
• The ability to make a measurable impact on critical trading infrastructure and strategies.
• A highly competitive compensation package, including:
  • Participation in a global profit-sharing pool and performance-based bonuses
  • 401(k) match up to 50%
  • 25 paid vacation days in addition to market holidays
  • Comprehensive health, dental, vision, disability, and life insurance coverage
• Extensive office perks, such as daily breakfast, lunch, snacks, regular social events, clubs, and sports leagues.

How do you apply?

If you are interested in applying for the Machine Learning Performance Engineer role, please do so via the link on this page or contact Digital Republic by phone or email. 

Contact us to find out more:

Get in contact with Digital Republic Talent by sending an email to [email protected]. Check out the website at www.digitalrepublictalent.com. You can also find out more on Linkedin, Instagram or Facebook.