Software Dev Engineer II, Stores Foundational AI -SFAI

What you'll do

• Design and build scalable training and RL post-training systems, pipelines, and data infrastructure.
• Improve training stability, efficiency, and reliability by optimizing compute and eliminating system bottlenecks.
• Partner with scientists and engineers to productionize algorithms, run experiments, and advance the technical roadmap.

What they ask for

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
• Experience programming with at least one software programming language.
• Knowledge of Machine Learning and LLM fundamentals, including transformer architecture, training/inference lifecycles, and optimization techniques.

Nice to have

• Knowledge of ML frameworks including JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, and TensorRT.
• Knowledge of system performance, memory management, and parallel computing principles; experience with CUDA/C++/Kernel development.

Pay

143,700.00 - 194,400.00 USD annually (base salary)

Tools

JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, TensorRT, CUDA/C++/Kernel

AI focus

Build infrastructure for LLM training and generative AI for shopping, including reinforcement-learning post-training pipelines and scalable production systems.