Founding ML Engineer

About the role

We’re looking for someone who loves open-source and collaborating with the OSS community but also has deep understanding in how transformers and Unsloth works.

Responsibilities

• Deep knowledge and implementation experience with multi-GPU training techniques such as FSDP
• Important contributions to ML open-source projects (e.g., transformers, TRL, kernels, Unsloth) with reviewable code
• Heavy experience working with ML frameworks like Unsloth, Transformers, TRL, and PyTorch
• Familiarity with inference engines such as vLLM
• Familiarity with low-level languages and kernels like Triton
• Understanding of cloud services such as GCP or AWS (bonus)

Requirements

• MUST COMPLETE CHALLENGES BEFORE APPLYING: https://x.com/danielhanchen/status/1891194528931209644
• Must have:
• - Deep knowledge and implementation experience with multi-GPU training techniques such as FSDP
• - Important contributions to ML open-source projects (e.g., transformers, TRL, kernels, Unsloth) with reviewable code
• - Heavy experience working with ML frameworks like Unsloth, Transformers, TRL, and PyTorch
• - Familiarity with inference engines such as vLLM
• - Familiarity with low-level languages and kernels like Triton
• - Understanding of cloud services such as GCP or AWS (bonus)

Qualifications

• Any (new grads ok)

Skills

• Amazon Web Services (AWS)
• Microsoft Azure
• Google Cloud
• Mathematica
• Torch/PyTorch
• Machine Learning
• Deep Learning
• Reinforcement learning (RL)

Benefits

Founded in San Francisco, Unsloth AI (YC S24) is backed by top-tier VCs and angels like Logan Kilpatrick (Lead Product at Google AI), Cliff Obrecht (Canva co-founder) and Jon Oringer (Shutterstock founder).

Pay

$200K - $500K

Schedule

0.30% - 0.70%

Location

San Francisco, CA, US