Machine Learning Performance Engineer in London

Machine Learning Performance Engineer in London

London Full-Time 80000 - 100000 Β£ / year (est.) No working from home possible
X

At a Glance

  • Tasks: Boost ML performance and AI acceleration in a cutting-edge tech environment.
  • Company: Join XTX, a leader in innovative technology with a dynamic team.
  • Benefits: Enjoy free meals, gym access, extensive medical benefits, and generous holiday.
  • Other info: Work in a vibrant Kings Cross office with excellent career growth.
  • Why this job: Tackle complex challenges and shape the future of AI technology.
  • Qualifications: Proficient in C/C++/Rust with experience in GPUs or TPUs.

The predicted salary is between 80000 - 100000 Β£ per year.

The Role

XTX is seeking an experienced engineer to support our ML Performance and AI Acceleration function.

You will be working in a small and highly technical team at extremely high- and low-levels of abstraction to dramatically increase the performance of XTX's training and inference platforms.

The remit is wide, and you should expect to be challenged.

We are not just writing CUDA code.

Amongst other things, you will be working on a sophisticated optimising compiler for a variety of accelerated computing platforms.

You should be proficient in memory hierarchies, degrees of parallelism, microarchitectural nuances on instruction dispatch rates and irregular bitwidth numerics.

You will also be expected to assist in the design of novel hardware architectures for accelerating our models, mapping AI models from JAX graphs to transistors.

Attributes

You will be an experienced software engineer, likely in C/C++/Rust or a similar language.

You will be confident solving complex algorithmic problems.

In addition, the following experience would be of particular interest

  • Understanding of, and some experience targeting, GPUs, TPUs, or similar hardware.
  • Some experience working with data-flow oriented compilers for AI or scientific computing.
  • Writing Linux kernel modules or other experience in OS development.

Benefits

Too many to list! Key benefits below and full benefits listed on our website.

  • Onsite gym, sauna, and fitness classes at no charge.
  • Extensive medical benefits including an on-site doctor and therapist at no charge.
  • Breakfast and lunch provided daily.
  • Various supports for caregivers, including emergency dependent care.

β€’ Beautiful Kings Cross office

  • 25 days paid holiday per year + statutory holiday and paid sick days.
  • Generous Pension contributions.
  • #J-18808-Ljbffr
X

Contact Details:

Xtxmarketstechnologies Recruitment Team

We think you need these skills to ace Machine Learning Performance Engineer in London

Machine Learning
CUDA Programming
C/C++/Rust
Algorithmic Problem Solving
Memory Hierarchies
Parallel Computing
Microarchitecture Knowledge