AI Training Systems Engineer: Distributed & RL

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

Benefits offered by this job

Comprehensive medical, dental, and vision insurance

Fully paid parental leave

Daily lunches and dinners provided

Relocation support and paid time off

Job summary

B Capital in San Francisco is looking for an engineering professional to architect and optimize core training infrastructure for their AI models. You will work on distributed systems and large-scale data pipelines, focusing on performance and numerical stability. Successful candidates will have strong software engineering skills and experience in either distributed training or data infrastructure. The role offers top-tier compensation and comprehensive health and wellness benefits.

Qualifications

• Strong software engineering skills with knowledge of machine learning.
• Experience in distributed systems and performance optimization.
• Ability to implement research papers into scalable systems.

Responsibilities

• Designing and optimizing large-scale training loops and data pipelines.
• Implementing state-of-the-art techniques ensuring numerical stability.
• Building internal tooling for launching and monitoring experiments.

Skills

Distributed Training & Inference

Data Infrastructure

Tools

PyTorch

JAX

Kubernetes

Ray

Job description

B Capital in San Francisco is looking for an engineering professional to architect and optimize core training infrastructure for their AI models. You will work on distributed systems and large-scale data pipelines, focusing on performance and numerical stability. Successful candidates will have strong software engineering skills and experience in either distributed training or data infrastructure. The role offers top-tier compensation and comprehensive health and wellness benefits.