
Member of Technical Staff, Training Infra at Inception (San Francisco Bay Area, CA)
Inception· San Francisco Bay Area, CA· $200k–$350k·
Role details
Salary
$200k–$350k
Work type
Onsite
Employment
Full-Time
Equity
Yes
Job description
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.
Why work at Inception
- Culture: A tight‑knit team of scientists, engineers, and builders focused on shipping frontier AI research. Emphasis on innovation, speed, and real‑world impact.
- Work policy: In‑office for most roles (Palo Alto HQ); a Marketing Intern position is listed as remote. Typical time on‑site is full‑time in the office.
- Notable perks / engineering culture: Opportunity to work on cutting‑edge diffusion LLMs from scratch, contribute to foundational research (papers published), and collaborate with alumni from top AI labs. Roles span from kernel engineering to RL infrastructure and product management.
- Growth: Rapidly scaling headcount (+187% YoY) with open positions across multiple disciplines, offering strong career progression in a high‑visibility startup.