Senior Staff Engineer to help define and build the AI Factory Reference Architecture, a Kubernetes-native platform for developing, training, evaluating, and deploying next-generation AI systems across internal and external customer environments.
Pay within range listed + Bonus + Equity
- Lead the design and implementation of the AI Factory Reference Architecture, delivering a Kubernetes-native platform for AI development, distributed training, simulation, evaluation, and deployment.
- Partner directly with ML researchers to understand evolving training workflows and ensure the platform supports state-of-the-art AI frameworks, foundation model development, reinforcement learning, distributed training, and emerging research workflows.
- Design self-service AI development workflows that enable engineers to move seamlessly from local experimentation to large-scale distributed execution using familiar open source tools and frameworks.
- Build the infrastructure required to support distributed training, simulation, inference, and reinforcement learning workloads.
- Design and optimize shared GPU infrastructure across cloud and on-premises environments.
- Nice to have: Ray
- Nice to have: KAI
- Nice to have: Slurm
- Nice to have: reinforcement learning
- Nice to have: simulation-driven training
Staff Engineer, Test AutomationNewf7d8749a-72d6-407c-bf98-292ea630c73a · San Mateo, US