LLM Training Dataset and Checkpoint Optimization Engineer
- Full-time
About Us
Together.ai is a leader in developing AI infrastructure that powers the training of state-of-the-art models. We focus on creating scalable, efficient systems for handling massive datasets and managing large-scale distributed checkpoints, ensuring seamless workflows for training and fine-tuning AI models.
We are seeking a Training Dataset and Checkpoint Acceleration Engineer to optimize data pipelines and checkpoint mechanisms for large-scale machine learning workloads. In this role, you will work at the intersection of data engineering and distributed systems, ensuring that training workflows are highly performant, reliable, and cost-efficient.
Responsibilities
- Dataset Acceleration :
Design and optimize high-throughput data pipelines for streaming and processing massive training datasets.
Build and optimize distributed checkpoint mechanisms for large-scale training workflows.
Profile and debug bottlenecks in data pipelines and checkpoint systems.
Develop systems that scale efficiently across thousands of nodes and petabyte-scale datasets.
Work closely with ML researchers, data engineers, and infrastructure teams to understand workload requirements.
Qualifications
Must-Have :
5+ years of experience in data engineering, distributed systems, or ML infrastructure.
Expertise in high-performance data processing libraries (e.g., PyTorch DataLoader, TensorFlow Data, DALI).
Proficient in Python, C++, or Go for performance-critical systems.
Experience with I / O optimization techniques (e.g., asynchronous data loading, prefetching).
Analytical and problem-solving mindset.
Nice-to-Have :
About Together AI
Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers in our journey in building the next generation AI infrastructure.
Compensation
We offer competitive compensation, startup equity, health insurance and other competitive benefits. The US base salary range for this full-time position is : $160,000 - $230,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.
Equal Opportunity
Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.
Please see our privacy policy at https : / / www.together.ai / privacy
#J-18808-Ljbffr
Create a job alert for this search
Llm And Optimization • San Francisco, CA, United States
Machine Learning Engineer, Distributed Training, Optimus
Applied ML / LLM Engineer
LLM Training Resilience Engineer
Police Officer - New Recruit (Entry Level)
Machine Learning Engineer | Python | Pytorch | Distributed Training | Optimisation | GPU | Hybrid, San Jose, CA
LLM Training Frameworks and Optimization Engineer
Senior LLM Engineer
IRL Tech III
Product Development Engineer, Reagents
LLM Training Frameworks and Optimization Engineer
Staff ML Platform Engineer – Large Scale Training (LLMOps / MLOps)
Lab Systems Training & Adoption Lead
Training : ML Framework Engineer
Machine Learning Engineer - Post Training
Tech Lead Manager- MLRE, ML Systems
Lecturer - Information and Cybersecurity - School of Information
Qualified Moms Wanted : Healthy Prior Delivery + Age 21–36 (Surrogacy $50k–$100k)
