Senior Machine Learning Performance Engineer
Locations: Austin, New York, Chicago
Our client, a world-class tech-driven trading firm, is in the process of assembling an advanced Machine Learning/AI team. We are seeking a Senior-level Machine Learning Performance Engineer for a critical role, leading an AI initiative with substantial influence on the firm’s trading systems and strategies.
About the Company:
Our client is a global leader in market-making, dedicated to driving transparency and liquidity across financial markets worldwide. Their Austin office is a hub for innovation, with a focus on machine learning, big data, and cutting-edge research infrastructure. As part of their AI team, you’ll be working in a collaborative environment with access to world-class technology and professionals.
Key Responsibilities:
- Architect and develop scalable machine learning pipelines for training and inference.
- Dive deep into the mechanics of deep learning frameworks (such as PyTorch, TensorFlow, JAX) and improve their efficiency.
- Collaborate with engineers and researchers to address and solve performance bottlenecks.
- Build a comprehensive understanding of complex trading systems to help optimize the performance of large-scale, low-latency infrastructure.
Essential Qualifications:
- Advanced expertise in deep learning frameworks (e.g., PyTorch, TensorFlow, JAX) and deep knowledge of computer architecture.
- Proficiency in programming languages like C++ and Python.
- Experience with large-scale distributed training and optimization.
Preferred Skills:
- Knowledge of the JAX ecosystem (XLA, Flax, etc.).
- Expertise in GPU programming and hardware accelerators (CUDA, Triton, etc.).
- Contributions to open-source machine learning or data science projects.
Benefits and Rewards:
- Industry-leading compensation packages with potential earnings far above the norm.
- Performance-based bonuses, a collaborative team environment, and global profit-sharing.
- Comprehensive benefits including 25 paid vacation days, fully paid health insurance, 401(k) matching, and more.
Do you have the passion and technical depth to drive cutting-edge machine learning solutions in a fast-paced environment? Join our client’s top-tier AI team and be part of an exciting journey with a direct impact on trading strategies.
Interested in applying? Get in touch with us today!
