poolside is hiring for a

Member of Engineering, GPU

Position: Full-time, indefinite contract

Location: Remote EMEA/East Coast or Paris

Apply for this role

About poolside

In this decade, the world will create artificial intelligence that reaches human level intelligence (and beyond) by combining learning and search. There will only be a small number of companies who will achieve this. Their ability to stack advantages and pull ahead will determine who survives and wins. These companies will move faster than anyone else. They will attract the world's most capable talent. They will be on the forefront of applied research and engineering at scale. They will create powerful economic engines. They will continue to scale their training to larger & more capable models. They will be given the right to raise large amounts of capital along their journey to enable this.

poolside exists to be one of these companies - to build a world where AI will drive the majority of economically valuable work and scientific progress.

We believe that software development will be the first major capability in neural networks that reaches human-level intelligence because it's the domain where we can combine Search and Learning approaches the best.

At poolside we believe our applied research needs to culminate in products that are put in the hands of people. Today we focus on building for a developer-led increasingly AI-assisted world. We believe that current capabilities of AI lead to incredible tooling that can assist developers in their day to day work. We also believe that as we increase the capabilities of our models, we increasingly empower anyone in the world to be able to build software. We envision a future where not 100 million people can build software but 2 billion people can.

About our team

We are a remote-first team that sits across Europe and North America and comes together once a month in-person for 3 days and for longer offsites twice a year.

Our R&D and production teams are a combination of more research and more engineering-oriented profiles, however, everyone deeply cares about the quality of the systems we build and has a strong underlying knowledge of software development. We believe that good engineering leads to faster development iterations, which allows us to compound our efforts.

About the role

Performance-oriented programming in CUDA, C++, Cython, Triton. You are going to accelerate high-level primitives used to train Large Language Models (LLMs) and optimize distributed communication over NVLink and InfiniBand. You will be working on poolside's own implementation of distributed training for LLMs.

Your mission

Ensure cutting-edge performance of LLM pre-training and fine tuning on huge state-of-the-art GPU clusters.

Responsibilities

Profile CPU and CUDA code at several abstraction levels.
Debug and profile distributed applications.
Troubleshoot undocumented CUDA internals.
Hack the NCCL library used for GPU communication.
Tune vanilla CUDA, Triton, CUTLASS kernels for the latest NVIDIA GPUs.
Hack PyTorch internals.
Deep dive into InfiniBand RDMA, including SHARP.

Skills & Experience

Engineering background
Expert understanding of GPU hardware/architecture
Strong C/C++ programming skills
Fine-grained knowledge of CUDA programming
Strong algorithmic skills
System programming on Linux experience
Plus: knowledge of CPython internals and experience of native extension development
Plus: knowledge of InfiniBand internals
Plus: compiler development background