About the Company
Adaptive ML is building a reinforcement learning platform to tune, evaluate, and serve specialized language models. We are pioneering the development of task-specific LLMs using synthetic data, creating the foundational tools and products needed for models to self-critique and self-improve based on simple guidelines.
- Location: New York / Paris
- Funding: Closed a $20M seed with Index & ICONIQ in early 2024
- Clients: Live with our first enterprise customers (e.g., AT&T)
Our Technical Staff develops the foundational technology that powers Adaptive ML in alignment with requests and requirements from our Commercial and Product teams. We are committed to building robust, efficient technology and conducting at-scale, impactful research to drive our roadmap and deliver value to our customers.
About the Role
Job Title: GPU Performance Engineer
Employment Type: Full-time
Department: Technical Staff
Location: Paris or New York (in-person role)
As a GPU Performance Engineer, you will help ensure that our LLM stack (Adaptive Harmony) delivers state-of-the-art performance across a variety of settings, from latency-bound regimes to throughput-bound regimes during training and offline inference. You will contribute to building the foundational technology powering Adaptive ML by delivering performance improvements directly to our clients and internal workloads.
Your Responsibilities
- Build and maintain fast and robust GPU code, focusing on delivering performance improvements in real-world applications.
- Write high-quality software in CUDA, CUTLASS, or Triton with a focus on performance and robustness.
- Profile dedicated GPU kernels, optimizing across latency/compute-bound regimes for complex workloads.
- Contribute to our product roadmap by identifying promising trends that can improve performance.
- Report clearly on your work to a distributed collaborative team, with a bias for asynchronous written communication.
Your (Ideal) Background
- M.Sc. / Ph.D. in computer science, or demonstrated experience in software engineering, preferably with a focus on GPU optimization.
- Strong programming skills, preferably with a focus on systems and general-purpose GPU programming.
- A track record of writing high-performance kernels, preferably demonstrating the ability to reach state-of-the-art performance on well-defined tasks.
- Contributions to relevant open-source projects, such as CUTLASS, Triton, and MLIR.
- Passionate about the future of generative AI, eager to build foundational technology to help machines deliver more singular experiences.
Benefits
- Comprehensive medical (health, dental, and vision) insurance
- 401(k) plan with 4% matching (or equivalent)
- Unlimited PTO — we strongly encourage at least 5 weeks each year
- Mental health, wellness, and personal development stipends
- Visa sponsorship if you wish to relocate to New York or Paris
