About the job
Our client, a leading global trading and technology firm, is seeking a Machine Learning Performance Engineer to join their team. This role is part of a key AI initiative and offers the opportunity to have a significant impact on machine learning infrastructure, addressing training and inference challenges to enhance their trading strategies.
Who they are
This technology-driven trading firm is recognized as a major player in the global market-making industry, partnering with over 70 exchanges worldwide. Their mission is to improve market efficiency by providing liquidity, accurate pricing, and transparency, while acting as a stabilizing force in various market conditions. With a continuous focus on innovation, they work toward safeguarding healthy and efficient markets for all participants.
What you'll do
- Build scalable and robust training and inference pipelines for deep learning.
- Dive deep into open-source deep learning frameworks and enhance their performance.
- Identify and eliminate performance bottlenecks in systems.
- Collaborate with researchers and other engineering teams.
- Gain a comprehensive understanding of the firm’s trading systems and strategies.
What you’ll need
- Expertise in deep-learning frameworks like PyTorch, JAX, TensorFlow, etc.
- A deep understanding of computer architecture.
- Proficiency in C++ and Python.
Nice to have
- Experience with the JAX ecosystem (XLA, Flax, etc.).
- Expertise 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 in data science and machine learning.
What you’ll get
- Collaborate with a global team of professionals from over 40 countries.
- Performance-based bonuses that are unmatched in the industry, with profits shared globally across teams, fostering a collaborative work environment.
- Opportunities to lead initiatives that directly impact business outcomes.
Additional benefits include 25 paid vacation days, market holidays, fully paid health insurance, daily meals, 401(k) matching up to 50%, charitable match opportunities, training programs, regular social events, and more.
The firm is dedicated to building a diverse and inclusive environment based on mutual respect. Recruitment, employment, training, compensation, and promotion decisions are made without regard to race, religion, color, sex, gender identity, sexual orientation, age, disability, or other legally protected characteristics.