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: https://vimeo.com/257888726
- 25 days paid holiday per year + statutory holiday and paid sick days.
- Generous Pension contributions.
