Member of Technical Staff - ML Training Infrastructure
Tasks
- Build GPU data pipelines
- Build evaluation harness for checkpoints
- Design fault tolerant long running training
- Ensure experiment tracking and reproducibility
- Implement checkpointing and restart
- Own distributed training end to end
- Profile and eliminate performance bottlenecks
- Set up observability and alerting
- Translate research requirements into production systems
Perks/Benefits
- N/A
Skills/Tech-stack
Alerting | CUDA | Checkpointing | Data Pipelines | Data loading | Distributed Training | Experiment tracking | Fault Tolerance | GPU Computing | Kubernetes | Machine Learning | NCCL | Networking | Observability | Orchestration | Profiling | PyTorch | Python | Ray | Reproducibility | Slurm | Storage Systems
Education
N/A
Regions
Countries
States
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R5d ago
-
Featured Feat. Associate Director, Data Labs USD 167K-167KAWS | Cloud Computing | Compute Infrastructure | Data Analysis | LLM GovernanceConference speaking opportunities | Hybrid work schedule | Media appearancesSenior-level Full TimeWashington, District of Columbia, 20004, United … R18d ago
-
Cloud Data Engineer II USD 107K-150KAWS | AWS Glue | AWS Step Functions | Amazon Athena | Amazon DynamoDBMid-level Full TimeMiami Lakes, FL, United States8h ago
-
Supply Chain Applied AI Engineer USD 200K-300KAI architecture | API Integration | Data Connectivity | Generative AI | Incident ResponseSenior-level Full TimeAuburn Hills, MI, United States10h ago
-
Senior Data Engineer USD 130K-204KApache Spark | Data Governance | Data Lineage | Data Modeling | Data PipelinesSenior-level Full TimeAuburn Hills, MI, United States10h ago
-
Senior-level Full TimeMiami11h ago
-
Senior-level Full TimeSan Francisco11h ago
-
AI Engineer (Agents) USD 160KAI SDK | Adversarial Testing | Arize Phoenix | Asynchronous programming | CI/CDSenior-level Full TimeNew York11h ago
-
Agentic AI Engineer USD 101K-151KAgentic Workflows | Debugging | Embeddings | Git | GitHubFinancial benefits | Health benefits | Wellness benefitsMid-level Full TimePennsylvania, Canonsburg12h ago
-
AI | AI Inference | CPU | Containers | GPUSenior-level Full TimeSeattle, Washington, United States14h ago
-
Senior Software Engineer, Cross Platform Applications USD 212K-387KArtificial Intelligence | Automation | CI/CD | Code Analysis | Dynamic analysisSenior-level Full TimeSan Jose, California, United States14h ago
-
Machine Learning Engineer - AI Compiler Optimization USD 156K-387KCo-design | Code generation | Compiler optimization | Graph optimization | Hardware-Software Co-designMid-level Full TimeSan Jose, California, United States14h ago
-
Cross-Functional Collaboration | Cross-functional | Data Pipelines | Full Stack | Full Stack Machine LearningSenior-level Full TimeSan Jose, California, United States14h ago
-
A/B | A/B Testing | B testing | Behavior Modeling | ExperimentationSenior-level Full TimeSan Jose, California, United States14h ago
-
Software Engineer (Leadership) - Infrastructure USD 160K-210KAI workflow | AI workflow automation | Agent Orchestration | Bias Mitigation | C plus plusSenior-level Full TimeSunnyvale, CA | Bellevue, WA | …15h ago
-
Software Engineer, Sensors & Computer Vision USD 146K-208K3D Reconstruction | C++ | Camera | Clustering | Convolutional Neural NetworksMid-level Full TimeSunnyvale, CA15h ago
-
C++ | Computer Vision | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA15h ago
-
Principal Engineer, AI Ecosystem USD 307K-428KCUDA | Cloud Native | Cloud Orchestration | Deep learning | Distributed SystemsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA15h ago
-
AI Engineer, Advanced Solutions Lab, Google Cloud USD 152K-222KApache Beam | Apache Hive | Artificial Intelligence | C++ | Cloud ComputingSenior-level Full TimeChicago, IL, USA; Atlanta, GA, USA15h ago
-
Senior Software Developer, Computer Vision, XR USD 174K-253KAugmented Reality | C++ | Computer Vision | Data Processing | DebuggingSenior-level Full TimeSan Jose, CA, USA; Waterloo, ON, …15h ago
-
Bi-Encoder | Cross-Encoder | Data Processing | Data Structures | Data Structures and AlgorithmsBonus | Equity | Health benefitsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA15h ago
-
Software Engineer III, AI/ML, gUP Customer Support USD 147K-211KAgile | C++ | Code review | Data Processing | DebuggingSenior-level Full TimeSunnyvale, CA, USA15h ago
-
Business Data Scientist, Applied Machine Learning, GCS USD 138K-198KA/B | A/B Testing | B testing | Causal Inference | Causal MLMid-level Full TimeMountain View, CA, USA; New York, …15h ago
-
Staff Software Engineer, ML Data Infrastructure, YouTube USD 207K-301KAlgorithms | C++ | Data Infrastructure | Data Processing | Data StructuresSenior-level Full TimeSan Bruno, CA, USA15h ago
-
AWS Data Engineering Advisor USD 117K-179KAWS | AWS Batch | AWS Glue | AWS Secrets | AWS Secrets Manager401k retirement savings plan | ESPP | Employer-paid disability insurance | Health, dental, and vision insurance | HolidaysSenior-level Full TimeHOUSTON, US, 7705617h ago