Post-Training AI Research Engineer, Distributed Systems

The Consensus; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

100% coverage of medical, dental, and vision insurance

Flexible PTO policy

Company-facilitated 401(k)

Meaningful equity

Job summary

The Consensus is seeking a Research Engineer to join its Post-Training team in San Francisco. The ideal candidate will have strong experience in machine learning and be proficient in tools like PyTorch and TensorFlow. You will be responsible for building tooling to support model training at scale, ensuring efficiency and quality for our customers.

We offer a competitive compensation package, including equity, medical coverage, and a flexible PTO policy. Join us in shaping the future of AI with a diverse and inclusive team.

Qualifications

• Strong experience in machine learning and solid foundations in mathematics and computer science.
• Ability to work with various model architectures and techniques efficiently at scale.

Responsibilities

• Build in-house tooling to support model training at scale.
• Work across technical stacks including Kubernetes and GPU computations.

Skills

Deep understanding of modern ML techniques

Advanced experience in tensor computation libraries

Understanding of transformer training parallelism strategies

Experience profiling distributed GPU programs

Familiarity with HPC platforms like Slurm and Kubernetes

Tools

PyTorch

TensorFlow

Jax

Job description

The Consensus is seeking a Research Engineer to join its Post-Training team in San Francisco. The ideal candidate will have strong experience in machine learning and be proficient in tools like PyTorch and TensorFlow. You will be responsible for building tooling to support model training at scale, ensuring efficiency and quality for our customers.

We offer a competitive compensation package, including equity, medical coverage, and a flexible PTO policy. Join us in shaping the future of AI with a diverse and inclusive team.