Senior ML Engineer: Foundation Models & Inference

Jobgether; España; Híbrido; EUR 90.000 - 130.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Competitive pay

Flexible work environment

Learning opportunities

Global team

Impactful projects

Descripción de la vacante

Partner Company is seeking a Senior ML Engineer (Token Factory) based in Spain to lead large-scale ML infrastructure projects. You will build inference and fine-tuning technologies for foundation models across language, vision, audio, and multimodal architectures.

You will collaborate with experienced engineers and researchers in a fast-moving, international environment, turning research ideas into production-ready systems with strong software practices.

Formación

• Strong foundation in machine learning and reinforcement learning.
• Experience training large ML models across multiple nodes.
• Proficiency with Python and modern DL frameworks (JAX).
• Exposure to CI/CD, version control, testing, and production-grade software engineering.

Responsabilidades

• Develop and improve fine-tuning methodologies for foundation models including LoRA and full-parameter approaches.
• Optimize model quality and training efficiency at large scale.
• Address bottlenecks in LLM inference for production efficiency.
• Build training/evaluation pipelines using JAX (speculative decoding, advanced inference).
• Experiment with model architectures (dense, mixture-of-experts, autoregressive, parallel).
• Develop scaling laws to guide model development and resource allocation.
• Investigate low-precision training/inference (FP8, NVFP4, MXFP4).
• Work with distributed training across multiple GPUs/nodes.

Conocimientos

ML fundamentals

Reinforcement learning

Distributed training

Python

JAX

CI/CD

English communication

Research to production

Herramientas

JAX

Python

Descripción del empleo

Partner Company is seeking a Senior ML Engineer (Token Factory) based in Spain to lead large-scale ML infrastructure projects. You will build inference and fine-tuning technologies for foundation models across language, vision, audio, and multimodal architectures.

You will collaborate with experienced engineers and researchers in a fast-moving, international environment, turning research ideas into production-ready systems with strong software practices.