Sr Software Dev Engineer, Stores Foundational AI -SFAI
Core
Architect and build scalable ML infrastructure for training and deploying large language models to power AI-driven shopping experiences.
Role type
Senior Software Development Engineer (ML Infrastructure)
Builds
Scalable RL post-training pipelines, experimentation frameworks, and production-ready systems for LLMs
Domain
E-commerce, Generative AI, Reinforcement Learning
Deliverable
production ML models
Required skills
Distributed systems design, ML infrastructure, RL algorithms (PPO/GRPO/RLOO), GPU optimization, System observability, Cross-functional leadership
Preferred skills
JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, TensorRT, Transformer architecture, Big data technologies
Technologies
JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, TensorRT, GPU clusters
Responsibilities
Design and build end-to-end RL post-training pipelines at cluster scale; Optimize RL training stability and efficiency; Profile and eliminate bottlenecks across compute, networking, and storage; Build observability systems for training dynamics; Mentor engineers and contribute to system design and long-term technical roadmap
Seniority
Senior, hands-on IC with technical leadership