Y-Axis Jobs Logo
Send us feedback
telephone  +91 7670 800 001
Log in Sign up Try Premium
Member of Technical Staff, RL Infra at INCEPTION ARTIFICIAL INTELLIGENCE L.L.C - O.P.C
San Francisco, California, United States -
Full Time

Start Date

Immediate

Expiry Date

08 Jun, 26

Salary

0.0

Posted On

10 Mar, 26

Experience

5 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

No

Skills

Reinforcement Learning, Infrastructure Design, System Optimization, Scalability, Efficiency, Rollout Pipelines, Reward Pipelines, Reliability, Observability, Orchestration, PyTorch, TensorFlow, Ray, Megatron, PPO, DPO

Industry

technology;Information and Internet

Description
The Role We're looking for engineers and scientists to design, optimize, and maintain the core systems that enable scalable, efficient reinforcement learning for large models. This role sits at the intersection of research and large-scale systems engineering: you'll wear many hats, from optimizing rollout and reward pipelines to enhancing reliability, observability, and orchestration, collaborating closely with researchers to make RL stable, fast, and production-ready. Key Responsibilities * Design, build, and optimize the infrastructure that powers large-scale reinforcement learning and post-training workloads. * Improve the reliability and scalability of RL training pipelines, distributed RL workloads, and training throughput. * Develop shared monitoring and observability tools to ensure high uptime, debuggability, and reproducibility for RL systems. Qualifications * BS/MS/PhD in Computer Science, Engineering, or a related field (or equivalent experience). * Understanding of ML frameworks (PyTorch, TensorFlow, Ray, Megatron) from a systems perspective. * Experience working with reinforcement learning workloads (PPO, DPO, RLHF, or reward modeling). * 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.
Responsibilities
The role involves designing, building, and optimizing the core infrastructure that supports large-scale reinforcement learning and post-training workloads. Key tasks include improving pipeline reliability, scalability, and developing shared monitoring tools for high uptime and reproducibility.