Machine Learning Engineer — Training Optimization
Tasks
- Build and maintain training infrastructure
- Collaborate with research teams on training strategies
- Define and improve training performance metrics
- Evaluate and integrate advanced training techniques
- Improve distributed training approaches
- Optimize large scale model training pipelines
- Profile and resolve performance bottlenecks
- Translate research concepts into production systems
- Tune training components
Perks/Benefits
- Continuous learning
- Cutting-edge AI work
- Equity opportunities
- Experimentation culture
- Fast feedback loops
- High ownership
- Technical quality focus
Skills/Tech-stack
BF16 | Backpropagation | Checkpointing | Convergence | Data parallelism | Deep learning | DeepSpeed | Distributed Training | FP16 | FP8 | FSDP | Fault Tolerance | GPU Performance | GPU Performance Optimization | GPU Training | Gradient Checkpointing | Learning Rate | Learning Rate Scheduling | Machine Learning | Megatron | Mixed Precision | Model Parallelism | Multi-GPU | Multi-GPU Training | Multi-node | Multi-node Training | Numerical precision | Optimization Algorithms | Performance optimization | Pipeline parallelism | Profiling | PyTorch | Reproducibility | Zero
Education
N/A
Related jobs
-
AI Agents | AI Foundry | Azure AI | Azure AI Foundry | Azure DevOpsFlexible working hours | Hybrid working | International working environment | Knowledge sharing sessions | Learning opportunitiesEntry-level InternshipZurich, Switzerland R16h ago
-
Attention Mechanisms | Batching | CUDA | Cache optimization | Compute GraphsEquity participation | Flexible remote workSenior-level Full TimeSwitzerland18h ago
-
APIs | Agent Development | Agent Development Kit | Agent systems | Computer VisionSenior-level Full TimeZürich, Switzerland19h ago
-
Access Transparency | Artificial Intelligence | C++ | Cloud APIs | Customer-managed encryption keysSenior-level Full TimeZürich, Switzerland19h ago
-
Software Engineer, Cloud AI Research, Co-Scientist CHF 122K-227KAgent systems | Agentic AI | Algorithms | C++ | Data AnalysisMid-level Full TimeZürich, Switzerland19h ago
-
Senior AI Engineer CHF 129K-192KBehavior analytics | Detection and Response | Fine Tuning | Langchain | Language ModelsAnnual training budget | Company events and workshops | Enhanced maternity leave | Enhanced paternity leave | Hybrid optionsSenior-level Full Timeremote or office, Zürich, Switzerland R20h ago
-
Cloud Computing | Code review | Data Modalities | Deep learning | DockerCollaborative & Inclusive Culture | Four-day work week | Fully remote work | Opportunity for open source contributions | Unlimited paid time offSenior-level Full TimeSwitzerland R22h ago
-
Deep learning | Distributed Training | Experiment tracking | Inference Optimization | JAXAsync-first culture | Equity opportunities | High ownership | Remote-friendlySenior-level Full TimeSwitzerland23h ago
-
Senior AI Engineer / Technical Consultant CHF 128K-192KAPI Development | Amazon Web Services | Azure | CI/CD | Cloud platformSenior-level Full TimeLausanne, Vaud, Switzerland1d ago
-
Senior-level Full TimeLausanne, Vaud, Switzerland1d ago
-
Embedded Software Engineer (Camera & Driver Platform) CHF 102K-138KBash | C++ | Cross-compilation | Cross-debugging | DNSCar parking | Enhanced maternity and paternity leave | Flexible working hours | Generous holiday allowance | Hybrid workplaceMid-level Full TimeZurich, Switzerland2d ago
-
Senior Data Scientist CHF 128K-192KAWS | Apache Spark | Azure | Databricks | GitCross-functional teamwork | Customer contact | Professional development | Travel requiredSenior-level Full TimePully, Switzerland2d ago
-
A/B | A/B Testing | AWS | Azure | B testingFully remote work | Healthcare coverage | Paid parental leave | Relocation assistanceMid-level Full TimeSwitzerland R4d ago
-
Software Engineer - Embedded Software (d/f/m) CHF 95K-146KAI Inference | C# | C++ | Git | JavaAccident insurance | Defined contribution pension plan | Global employee benefits | Sick pay insurance | Wellbeing initiativesMid-level Full TimeCH-ZH-ZURICH-TECHNOPARK STRASSE 1, Switzerland4d ago
-
Robotics Software Engineer - Sim-to-Real CHF 115K-158KC++ | Embedded Systems | Physics simulation | PyTorch | PythonComprehensive benefits packageMid-level Full TimeSwitzerland, Zurich4d ago
-
Research Scientist/ Engineer - Special Projects CHF 134K-192K3D Reconstruction | Computer Graphics | Computer Vision | Deep learning | Image SynthesisSenior-level Full TimeZurich7d ago
-
AWS | Apache Airflow | Cloud platform | Deep learning | DockerAccess to modern development tools | Equity opportunities | Medical/Dental/Vision insurance | Unlimited paid time off | Work from homeMid-level Full TimeSwitzerland R7d ago
-
AWS | Apache Spark | Data Analysis | Hadoop | JavaRemote workSenior-level Full TimeZürich, Switzerland R7d ago
-
Senior ML Engineer (Zürich) CHF 140K-232KAdversarial Testing | Computer Vision | Data Generation | Data pipeline | Generative AISenior-level Full TimeZürich8d ago
-
Staff ML Engineer (Zürich) CHF 140K-232KComputer Vision | Data Modeling | Data collection | Language Processing | Machine LearningSenior-level Full TimeZürich8d ago
-
Senior Expert I/II, Data Science, Modeling & Simulation CHF 102K-189KComputational Biology | Data Analysis | Data Mining | Data Visualization | Deep learningFlexible working options | Global recognition programs | Hybrid working | Insurance plans | Paid parental leaveSenior-level Full TimeBasel (City), Switzerland8d ago
-
CUDA | Continuous Deployment | Continuous integration | Imitation Learning | JAXMid-level Full TimeSwitzerland, Zurich8d ago
-
Autoscaling | CI/CD | CUDA | Containerization | Distributed SystemsFully remote | High ownership | Remote-first collaboration | Startup growth opportunities | Work flexibilitySenior-level Full TimeSwitzerland R8d ago
-
Apache Airflow | Artificial Intelligence | BI platforms | CRM Data Platforms | CRM dataCollaborative culture | Continuous learning | Exposure to cutting edge AI and automation | Globally distributed team | Professional growthSenior-level Full TimeSwitzerland9d ago
-
AWS | Asynchronous programming | Deep learning | FastAPI | GitHub CopilotAsynchronous work | Career growth | Company-provided hardware | Remote-first | Work-life balanceMid-level Full TimeSwitzerland10d ago