About the Team Training Runtime designs the core distributed machine-learning training runtime that powers everything from early research experiments to frontier-scale model runs. With a dual mandate to accelerate researchers and enable frontier scale, we’re building a unified, modular runtime that meets researchers where they are and moves with them up the scaling curve. Our work focuses on three pillars: high-performance, asynchronous, zero-copy tensor and optimizer-state-aware data movement; performant, high-uptime, fault-tolerant training frameworks (training loop, state management, resilient checkpointing, deterministic orchestration, and observability); and distributed process management for long-lived, job-specific and user-provided processes. We integrate proven large-scale capabilities into a composable, developer-facing runtime so teams can iterate quickly and run reliably at any scale, partnering closely with model-stack, research, and platform teams. Success for us is measured by raising both training throughput (how fast models train) and researcher throughput (how fast ideas become experiments and products). About the Role As a Training: ML Framework Engineer, you will work on improving the training throughput for our internal training framework, while enabling researchers to experiment with new ideas. This requires good engineering (for example designing, implementing, and optimizing state-of-the-art AI models), writing bug-free machine learning code (surprisingly difficult!), and acquiring deep knowledge of the performance of supercomputers. In all the projects this role pursues, the ultimate goal is to push the field forward. We’re looking for people who love optimizing performance, understanding distributed systems, and who cannot stand having bugs in their code. Since our training framework is used for large runs with massive numbers of GPUs, performance improvements here will have a large impact. This role is based in San Francisco, CA. We use a
Jobs in United States
Training in United States
15 active opportunities · Updated September 2026
Showing
15 jobs
Explore current training jobs across United States. Filter by work mode, employment type, experience, department, date posted and distance.
About the Team Training Runtime designs the core distributed machine-learning training runtime that powers everything from early research experiments to frontier-scale model runs. With a dual mandate to accelerate researchers and enable frontier scale, we’re building a unified, modular runtime that meets researchers where they are and moves with them up the scaling curve. Our work focuses on three pillars: high-performance, asynchronous, zero-copy tensor and optimizer-state-aware data movement; performant, high-uptime, fault-tolerant training frameworks (training loop, state management, resilient checkpointing, deterministic orchestration, and observability); and distributed process management for long-lived, job-specific and user-provided processes. We integrate proven large-scale capabilities into a composable, developer-facing runtime so teams can iterate quickly and run reliably at any scale, partnering closely with model-stack, research, and platform teams. Success for us is measured by raising both training throughput (how fast models train) and researcher throughput (how fast ideas become experiments and products). About the Role As a Training Performance Engineer, you’ll drive efficiency improvements across our distributed training stack. You’ll analyze large-scale training runs, identify utilization gaps, and design optimizations that push the boundaries of throughput and uptime. This role blends deep systems understanding with practical performance engineering — analyzing GPU kernel performance, collective communication throughput, investigating I/O bottlenecks, and sharding our models so we can train them at massive scale. You’ll help ensure that our clusters are running at peak performance, enabling OpenAI to train larger, more capable models with the same compute budget. This role is based in San Francisco, CA. We use a hybrid work model of three days in the office per week and offer relocation assistance to new employees. In this role, you will: Profil
ABOUT BASETEN Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F , led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products. We are looking for an engineer with strong experience in machine learning and solid foundations in maths and computer science to join our growing Post-Training team at Baseten. Custom models are instrumental to the success of Baseten customers. By inference volume, the overwhelming majority of traffic at Baseten is to and from models that have been post-trained in some way, whether that be through reinforcement learning, supervised finetuning, a recent technique from the literature, or an in-house research technique from Baseten. The Post-Training team is responsible for the success of our customers’ post-trained models, and we employ a wide array of techniques to produce models that are more efficient and higher quality than even the biggest closed source models for the customer’s specific needs. Your role as a research engineer is to build the in-house tooling to support all of this. We care about training a wide spectrum of different model architectures with a variety of techniques efficiently and at scale. At times this involves zooming deep into a particular technical topic, but more often if involves working across the stack as a whole - systems-level concepts like Kubernetes, cgroups, storage systems, and networking topologies, as well as PyTorch distributed tensor computation, and GPU kernels. RECENT RESEARCH Dense, on-policy or both? Repeated kv cache for long-running agents Distillation without the dark – rep
ABOUT BASETEN Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F , led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products. THE ROLE This role sits at the frontier of our research agenda. You will pursue open problems at the intersection of post-training methodology and performant inference, and then collaborate with research engineering to translate findings into production systems. A meaningful portion of your time will be dedicated to research that deepens our understanding of how models learn, alignment, and architectural efficiency — questions that may not have immediate product application. The remainder will be directed toward research that solves concrete problems for Baseten's platform and customers, who are the fastest growing AI companies in the world like Cursor, Lovable, and Notion. We are looking for someone with sharp research taste and genuine creative instinct for problem selection. Someone who can identify questions that matter, design clean experiments to answer them, and push the state of the art. The environment here is not theoretical, but rather research that can be validated with eager customers who are serving billions of tokens a second. RECENT RESEARCH Towards infinite context windows: neural KV cache compaction Dense, on-policy or both? Repeated kv cache for long-running agents Distillation without the dark – replicating black-box on-policy distillation on Baseten RESPONSIBILITIES Define and pursue a research agenda spanning both foundational and applied work, with the applied component connected to Baseten's pla
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. The Store Manager In Training (SMIT) job is a development focused role that is crafted to prepare you for a CVS Store Manager position. The SMIT program is designed to enhance your career as a future store leader. We support your development towards the store manager role by participating in a self-paced, collaborative training over the span of 16 weeks. We provide both virtual and hands on learning experiences to develop you as a leader. Our team of people, that support your growth toward your leadership career include: your District Leader, Human Resources Business Partner, Training Store Manager, and your Store Manager coach. Some of the skills needed for this role are: Communicate well verbally and in writing to support and lead your team. Perform customer care duties to provide high levels of service. Execute merchandising strategies to support store sales growth. Manage the store inventory and assets to maintain profitability. We are invested in your growth and anticipate that you will be as well. We expect our SMIT's to: Actively engage with your leader in Key Learning Experiences to support you on your path to promotion during the structured program. Support your store as management team member and lead in a manner that is consistent with CVS values and p
Other cities to consider
More places hiring for this role
Get new training jobs in United States by email
Daily job updates · Unsubscribe anytime