Machine Learning Engineer — Training Optimization
Tasks
- Build and maintain training infrastructure for checkpointing fault tolerance reproducibility
- Collaborate with researchers on architecture aware training strategies
- Evaluate and integrate training techniques
- Improve distributed training strategies
- Optimize large scale model training pipelines
- Optimize training precision
- Own training performance metrics and improve continuously
- Reduce training time and compute cost through profiling and bottleneck analysis
- Tune optimizers schedulers and batch sizing
Perks/Benefits
- N/A
Skills/Tech-stack
BF16 | Backpropagation | Batch sizing | Checkpointing | Data parallelism | DeepSpeed | Distributed Training | FP16 | FP8 | FSDP | Fault Tolerance | Gradient Checkpointing | Learning Rate | Learning Rate Scheduling | Megatron | Mixed Precision | Model Parallelism | Neural Networks | Optimization Algorithms | Optimizers | Pipeline parallelism | Profiling | PyTorch | Reproducibility | Zero
Education
N/A
Related jobs
-
AWS SageMaker | Airflow | Amazon EC2 | Amazon EKS | Computer VisionContinuous learning and development | Flexible schedule | Fully remote | Mentorship opportunitiesMid-level Full TimeIndia R1d ago
-
Senior Machine Learning Engineer AUD 148K-210KAccess Control | Agent systems | CI/CD | Capabilities Based Access Control | Error RecoveryHealth and wellness benefits | Hybrid work schedule | Remote work flexibilitySenior-level Full TimeAustralia - Melbourne - Hybrid R1d ago
-
Senior Machine Learning Engineer AUD 148K-210KACID | Distributed Systems | Durable Execution | Event Sourcing | Fault ToleranceHybrid work | Remote work flexibilitySenior-level Full TimeAustralia - Melbourne - Hybrid R1d ago
-
Senior-level Full TimeSingapore R1d ago
-
Lead Machine Learning Engineer CAD 156K-251KAWS | Azure | CI/CD | Cloud Computing | Cloud platformAutonomy | Career development support | Learning and development programs | Mentorship and coachingSenior-level Full TimeToronto, Canada R1d ago
-
Lead Machine Learning Engineer USD 172K-276KAmazon Web Services | Azure | CI/CD | Cloud platform | DatabricksLearning and development programs | Mentorship and coachingSenior-level Full TimeChicago, Illinois, USA R1d ago
-
Senior Applied Scientist USD 170K-286KApache Spark | C++ | Computer Vision | Data Processing | Edge inferenceComprehensive health insurance | Parental leave | Professional development stipend | Remote work | Travel up to 5 percentSenior-level Full TimeRemote - US R1d ago
-
Senior Machine Learning Engineer USD 140K-224KAIOps | AWS | Apache Spark | Cloud platform | Data Analysis401k match | Dental insurance | Flexible paid time off | Life insurance | Medical insuranceSenior-level Full TimeRemote, United States R1d ago
-
Associate AI-ML Engineer INR 3200K-4500KCI/CD | CrewAI | Databases | Docker | Generative AILearning and growth | Mentorship | Mentorship from technology leadersMid-level Full TimeBengaluru East, Karnataka, India R1d ago
-
Senior Machine Learning Engineer (Fraud) CAD 153K-213KAirflow | Apache Spark | Batch Processing | CatBoost | DaskDental coverage | Employee Stock Purchase Plan (ESPP) | Flexible spending wallets | Health coverage | Remote work optionSenior-level Full TimeRemote Canada R1d ago
-
Senior, Machine Learning Engineer - 3D Perception USD 177K-212K3D Perception | Camera-based perception | Cloud processing | Computer Vision | Data Preparation401k employer match | Accidental death and dismemberment insurance | Flexible schedule | Holiday office closures | Life insuranceSenior-level Full TimeRemote, U.S, Ann Arbor, MI, Fort … R1d ago
-
Asynchronous training | Containerization | Curriculum learning | Distributed Systems | Distributed TrainingSenior-level Full TimeSF Bay Area, CA, Remote, US, … R1d ago
-
Associate AI-ML Engineer INR 3200K-4500KCI/CD | CrewAI | Data Preparation | Databases | DockerCareer growth opportunities | MentorshipMid-level Full TimeBengaluru South, Karnataka, India R1d ago
-
Machine Learning Engineer — Distillation USD 175K-250KDeep learning | Distributed Training | GPU Training | JAX | Knowledge DistillationAsync-first environment | Meaningful equity | Remote-friendlySenior-level Full TimeRemote (world) R2d ago
-
Machine Learning Engineer — AI Architecture Research USD 180K-300KAttention Mechanisms | Compute tradeoffs | Deep learning | JAX | Latency optimizationEquity | Fast feedback loops | Meaningful impact on technical direction | Opportunity to ship research into production | Work on core model architectureSenior-level Full TimeRemote (world) R2d ago
-
Machine Learning Engineer — Inference Optimization USD 170K-287KBF16 | CUDA | Deep learning | Distributed Systems | FP16Senior-level Full TimeRemote (world) R2d ago
-
Machine Learning Engineer (Platform) USD 156K-211KAWS | CPU Optimization | Data Processing | Data throughput | Distributed TrainingRemote workSenior-level Full TimeRemote-US R2d ago
-
Senior-level Full TimeToronto R2d ago
-
AWS | Autoscaling | Bash | CI/CD | Data PipelinesFully remote work | Healthcare coverage | Paid parental leave | Relocation assistance | Technical ownershipMid-level Full TimeRomania R2d ago
-
A/B | A/B Testing | AWS | Alerting | AutoscalingFully remote within Europe | Healthcare coverage | Paid parental leave | Professional growth | Relocation assistanceMid-level Full TimeItaly R2d ago
-
AWS | Autoscaling | Azure | Bash | CI/CDFully remote within Europe | Healthcare coverage | Paid parental leave | Professional growth | Relocation assistanceMid-level Full TimePortugal R2d ago
-
AWS | Bash | Cloud platform | Databricks | DatadogEquity package | Fully remote Europe | Healthcare coverage | Paid parental leave | Relocation assistanceMid-level Full TimeNetherlands R2d ago
-
A/B | A/B Testing | AWS | Alerting | AzureFully remote Europe | Healthcare coverage | Paid parental leave | Professional growth | Relocation assistanceMid-level Full TimeIreland R2d ago
-
A/B | A/B Testing | AWS | Azure | B testingFully remote work | Healthcare coverage | Paid parental leave | Relocation assistanceMid-level Full TimeSwitzerland R2d ago
-
AWS | Autoscaling | Azure | Bash | CI/CDEquity package | Fully remote work within Europe | Healthcare coverage | Paid parental leave | Relocation assistanceMid-level Full TimeFrance R2d ago