Machine Learning Performance Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

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.

BankingEmploymentRecruiting
Hiring Manager
Abbey Milligan
linkedin

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

Deep learning frameworksC++ programmingPython programmingComputer architectureLinux system programmingDistributed trainingOpen-source contributionsGPU programming

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

company-logo
Global recruitment firm specialising in Banking

Funding

Current Stage
Late Stage
Company data provided by crunchbase