7mo ago

Member of Technical Staff, Training Infra

Bay Areafull-timelead

OtherSoftware EngineerMember Of Technical StaffSoftware Engineering

1 views0 saves0 applied

Apply Now

Quick Summary

Overview

The Role We're looking for engineers and scientists to design, optimize, and maintain the core systems that enable scalable, efficient training of LLM.

Technical Tools

OtherSoftware EngineerMember Of Technical StaffSoftware Engineering

The Role

We're looking for engineers and scientists to design, optimize, and maintain the core systems that enable scalable, efficient training of LLM. Your goal is to make experimentation and training at Inception fast and reliable so our team can focus on science, not system bottlenecks.

Key Responsibilities

Design, implement, and optimize distributed training systems that scale across thousands of GPUs and nodes.
Develop high-performance optimizations to maximize throughput and efficiency.
Develop reusable frameworks and libraries to improve training reproducibility, reliability, and scalability for new model architectures.

Qualifications

BS/MS/PhD in Computer Science, Engineering, or a related field (or equivalent experience).
Understanding of ML frameworks (PyTorch, TensorFlow) from a systems perspective.
Strong engineering skills — ability to contribute performant, maintainable code and debug in complex codebases.
Proficiency in Python and at least one systems programming language (C++/Rust/Go).
Experience with containerization (Docker), orchestration (Kubernetes), and CI/CD pipelines.

Preferred Skills

Experience building and maintaining large-scale language models with tens of billions of parameters or more.
Experience with ML workflow orchestration tools (Kubeflow, Airflow).
Background in performance optimization and profiling of ML systems (Prometheus, Grafana, OpenTelemetry).
Familiarity with distributed frameworks such as PyTorch/XLA, DeepSpeed, Megatron-LM.

What We Offer

~1 min read

The annual base salary range for this role is $200,000 – $350,000 USD. Final compensation is determined based on experience, skills, and qualifications. Equity and benefits are included in the total package.