This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a Machine Learning Engineer — Training Optimization based in Spain.

This role offers the opportunity to improve the foundations behind large-scale AI model development and deployment.
You will work at the intersection of machine learning research, systems engineering, and production optimization.
The position focuses on making model training faster, more stable, and more cost-efficient through advanced engineering techniques.
You will optimize training pipelines, improve distributed systems, and collaborate with researchers to push model capabilities forward.
This is a high-impact opportunity for an engineer who enjoys solving complex performance challenges at scale.
You will have significant ownership in shaping training infrastructure, experimentation workflows, and the future of AI systems.

Accountabilities

As a Machine Learning Engineer focused on Training Optimization, you will improve the efficiency, scalability,



and reliability of large-scale model training systems. You will combine deep technical expertise with practical engineering execution to optimize how advanced AI models are developed.
- Optimize large-scale model training pipelines to improve throughput, convergence, stability, and overall computational efficiency.
- Improve distributed training approaches, including data parallelism, model parallelism, and pipeline parallelism strategies.
- Tune key training components such as optimizers, learning rate schedulers, batch sizes, and numerical precision methods including bf16, fp16, and fp8.
- Identify and resolve performance bottlenecks through profiling, system analysis, and infrastructure-level improvements.
- Collaborate closely with research teams to develop architecture-aware training strategies and improve model performance.
- Build and maintain reliable training infrastructure, including checkpointing systems, fault

📌 Machine Learning Engineer — Training Optimization (España)
🏢 Jobgether
📍 España