Member of Engineering (Pre-training and inference fault tolerance)

ABOUT POOLSIDE

In this decade, the world will create Artificial General Intelligence. There will only be a small number of companies who will achieve this. Their ability to stack advantages and pull ahead will define the winners. 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, engineering, infrastructure and deployment at scale. 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. They will create powerful economic engines. They will obsess over the success of their users and customers.

poolside exists to be this company - to build a world where AI will be the engine behind economically valuable work and scientific progress.

View GDPR Policy

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

You would be working in our pre-training team focused on building out our distributed training and inference of Large Language Models (LLMs). This is a hands-on role that focuses on software reliability and fault tolerance. You will work on cross-platform checkpointing, NCCL recovery, and hardware fault detection. You will make high-level tools. You will not be afraid of debugging Linux kernel modules. You will have access to thousands of GPUs to test changes.

Strong engineering skills are a prerequisite. We assume good knowledge of Torch, NVIDIA GPU architecture, reliability concepts, distributed systems, and best coding practices. A basic understanding of LLM training and inference principles is required. We look for fast learners who are prepared for a steep learning curve and are not afraid to step out of their comfort zone.

YOUR MISSION

To help train the best foundational models for source code generation in the world

RESPONSIBILITIES

• Identify, study, and troubleshoot hardware problems during training at scale

• Minimize the GPU idle time during faults, both operationally and strategically

• Design and develop tools and add-ons to accelerate the training recovery

• Improve the performance and reliability of checkpointing

• Write high-quality Python (PyTorch), Cython, C/C++, CUDA API code

SKILLS & EXPERIENCE

• Understanding of Large Language Models (LLM)

  • Basic knowledge of Transformers

  • Knowledge of deep learning fundamentals

• Strong engineering background

• Programming experience

  • Linux API, Linux kernel

  • Strong algorithmic skills

  • Python with numpy, PyTorch, or Jax

  • C/C++

  • NCCL

  • Use modern tools and are always looking to improve

  • Strong critical thinking and ability to question code quality policies when applicable

• Distributed systems

  • Reliability

  • Observability

  • Fault-tolerance

  • K8s stack

PROCESS

• Intro call with one of our Founding Engineers

• Technical Interview(s) with one of our Founding Engineers

• Team fit call with the People team

• Final interview with Eiso, our CTO & Co-Founder

BENEFITS

• Fully remote work & flexible hours

• 37 days/year of vacation & holidays

• Health insurance allowance for you and dependents

• Company-provided equipment

• Wellbeing, always-be-learning and home office allowances

• Frequent team get togethers

• Great diverse & inclusive people-first culture

Data Retention

Except as otherwise permitted or required by applicable law or regulation, we will only retain your personal data for as long as necessary to fulfill the purposes we collected it for, as required to satisfy any legal, accounting, or reporting requirements, or as necessary to resolve disputes. To determine the appropriate retention period for personal data, we consider our statutory obligations, the amount, nature, and sensitivity of the personal data, the potential risk of harm from unauthorized use or disclosure of your personal data, the purposes we process your personal data for, and whether we can achieve those purposes through other means. We specify the retention periods for your personal data in our data retention policy. Under some circumstances we may anonymize your personal data so that it can no longer be associated with you. We reserve the right to use such anonymous and de-identified data for any legitimate business purpose without further notice to you or your consent. If you are offered and accept employment with poolside, the personal data we collected during the application and recruitment process will become part of your employment record, and we may use it in connection with your employment consistent with our employee personal data policies. If you do not become an employee, or, once you are no longer an employee of poolside, we will retain and securely destroy your personal data in accordance with our document retention policy and applicable laws and regulations.

Rights of Access, Correction, Erasure and Objection

It is important that the personal data we hold about you is accurate and current. Please keep us informed if your personal data changes during the recruitment process. By law you may have the right to request access to, correct, and erase the personal data that we hold about you, or object to the processing of your personal data under certain circumstances. You may also have the right to request that we transfer your personal data to another party. If you want to review, verify, correct, or request erasure of your personal data, object to the processing of your personal data, or request that we transfer a copy of your personal data to another party, please contact us at legal@poolside.ai . Any such communication must be in writing. We may request specific information from you to help us confirm your identity and your right to access, and to provide you with the personal data that we hold about you or make your requested changes. Applicable law may allow or require us to refuse to provide you with access to some or all of the personal data that we hold about you, or we may have destroyed, erased, or anonymized your personal data in accordance with our record retention obligations and practices. If we cannot provide you with access to your personal data, we will inform you of the reasons why, subject to any legal or regulatory restrictions.

Right to Withdraw Consent

Where you have provided your consent to the collection, processing, and transfer of your personal data, you may have the legal right to withdraw your consent under certain circumstances. To withdraw your consent, if applicable, contact us at legal@poolside.ai

Changes to this Privacy Notice

We reserve the right to update this Privacy Notice at any time, and we will provide you with a new Privacy Notice when we make any updates. If we would like to use your previously collected personal data for different purposes than those we notified you about at the time of collection, we will provide you with notice and, where required by law, seek your consent, before using your personal data for a new or unrelated purpose. We may process your personal data without your knowledge or consent only where required by applicable law or regulation.

Contact Us

If you have any questions about our processing of your personal data or would like to make an access or other request, please contact us at legal@poolside.ai