Selby Jennings ยท 4 weeks ago
Machine Learning Performance Engineer
Selby Jennings is a leading tech-driven trading firm seeking a Machine Learning Performance Engineer to join a pivotal AI initiative. This role offers the opportunity to make a significant impact on machine learning infrastructure, training, and inference challenges that directly support advanced trading strategies.
Responsibilities
Build scalable and robust training and inference pipelines for deep learning models
Dive into the internals of open-source deep learning frameworks and enhance their functionality
Identify and eliminate performance bottlenecks across systems
Collaborate closely with researchers and engineers to optimize workflows
Develop a deep understanding of trading systems and their integration with ML solutions
Qualification
Required
Expertise in the internals of deep learning frameworks such as PyTorch, JAX, TensorFlow, etc
Strong understanding of computer architecture
Proficiency in C++ and Python programming
Preferred
Experience with the JAX ecosystem (XLA, Flax, etc.)
Programming experience for GPUs or other accelerators (CUDA, Triton, Pallas, etc.)
Linux system programming skills
Familiarity with large-scale distributed training
Contributions to open-source projects in data science or machine learning
Benefits
25 paid vacation days plus market holidays
Daily breakfast and lunch
Training and development opportunities
Charitable match programs
Regular social events and clubs
Company
Selby Jennings
Global recruitment firm specialising in Banking
Funding
Current Stage
Late StageRecent News
Business Insider
2025-09-30
2025-07-10
Seattle TechFlash
2025-05-03
Company data provided by crunchbase