Python Machine Learning Engineer (Hybrid - Madrid)

UST España & Latam; Madrid; Presencial; EUR 60.000 - 90.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

23 days annual leave

Health care plan

Training platforms access

WFH/remote policy

Career growth opportunities

Descripción de la vacante

UST España & Latam is seeking a Machine Learning Engineer to design, build, and deploy enterprise-grade AI solutions in Madrid. You will work in a role requiring strong Python skills and hands-on TensorFlow/PyTorch experience, focusing on LLMs, AI agents, and RAG architectures.

Role involves three days per week on-site at client offices in central Madrid, with a high emphasis on collaboration across Data Scientists, Data Engineers, and Platform Teams. Fluent English is required.

Formación

• Bachelor’s or Master’s degree in Computer Science, Data Science, Mathematics or related field.
• Strong programming skills in Python.
• Hands-on experience with TensorFlow and/or PyTorch.
• Proven experience with Large Language Models (LLMs).
• Solid understanding of AI Agents, agentic workflows, orchestration frameworks, and reasoning patterns.
• Experience designing and implementing Retrieval-Augmented Generation (RAG) architectures.
• Knowledge of data preprocessing, feature engineering, model selection and evaluation techniques.
• Strong grounding in statistics, probability, linear algebra and optimization.

Responsabilidades

• Develop, deploy, and optimize ML and AI solutions for complex business challenges.
• Design multi-agent systems and enable AI models with function/tool-calling capabilities.
• Build and maintain RAG solutions using enterprise data.
• Evaluate, integrate, and optimize AI models, including LLMs.
• Design and improve agentic workflows for production environments.
• Test and optimize prompts and few-shot examples for accurate, safe outputs.
• Collaborate with Data Scientists, Data Engineers, Platform Teams and stakeholders.

Conocimientos

Python

TensorFlow

PyTorch

LLMs

AI Agents

RAG architectures

Data preprocessing

Feature engineering

Model evaluation

Statistics & math

Software engineering best practices

Educación

BSc/MSc in CS/Data Science/Math

Herramientas

Git

CI/CD

Descripción del empleo

We are still looking for the very Top Talent… and we would be delighted if you were to join our team!

More in details, UST is a multinational company based in North America, certified as a Top Employer and Great Place to Work company with over 35.000 employees all over the world and presence in more than 35 countries. We are leaders on digital technology services, and we provide large-scale technologic solutions to big companies.

What we look for?

A passionate Machine Learning Engineer with experience in Generative AI, Large Language Models (LLMs), AI Agents, and MLOps/DevOps environments, eager to help design, build, and deploy enterprise-grade AI solutions.

Required to go to the client's offices in central Madrid 3 days a week.

High english level is required.

• Develop, deploy, and optimize Machine Learning and AI solutions for complex business challenges.
• Design and implement multi-agent systems and enable AI models with function/tool-calling capabilities.
• Build and maintain Retrieval-Augmented Generation (RAG) solutions using enterprise data.
• Evaluate, integrate, and optimize AI models, including Large Language Models (LLMs).
• Design and improve agentic workflows and AI agents for production environments.
• Test and optimize prompts and few-shot examples to ensure accurate, consistent, and safe AI outputs.
• Evaluate model performance using appropriate metrics, benchmarks, and validation techniques.
• Collaborate with Data Scientists, Data Engineers, Platform Teams, and other stakeholders.
• Participate in code reviews, testing, debugging, and quality assurance activities.
• Contribute to the full AI lifecycle, from design and development to deployment and monitoring.
• Bachelor’s or Master’s degree in Computer Science, Data Science, Mathematics, Statistics, or a related field.
• Strong programming skills in Python.
• Hands‑on experience with Machine Learning frameworks such as TensorFlow and/or PyTorch.
• Proven experience working with Large Language Models (LLMs).
• Solid understanding of AI Agents, agentic workflows, orchestration frameworks, and reasoning patterns.
• Experience designing and implementing RAG architectures.
• Knowledge of data preprocessing, feature engineering, model selection, and evaluation techniques.
• Strong understanding of statistics, probability, linear algebra, and optimization concepts.
• Experience applying software engineering best practices, including version control, testing, and documentation.

Nice to have:

• Experience working in cloud environments (Azure, AWS, or GCP).
• Knowledge of MLOps and CI/CD practices.
• Experience deploying and monitoring AI solutions in production.
• Background in financial services, banking, or wealth management.
• Familiarity with vector databases and modern GenAI ecosystems.

¿What can we offer?

• - 23 days of Annual Leave plus the 24th and 31st of December as discretionary days!
• - Numerous benefits (Heath Care Plan, Internet Connectivity, Life and Accident Insurances).
• - Free access to several training platforms
• - Professional stability and career plans
• - UST also, compensates referrals from which you could benefit when you refer professionals.
• - The option to pick between 12 or 14 payments along the year.
• - Real Work Life Balance measures (flexibility, WFH or remote work policy, compacted hours during summertime…)
• - UST Club Platform discounts and gym Access discounts

In UST we are committed to equal opportunities in our selection processes and do not discriminate based on race, gender, disability, age, religion, sexual orientation or nationality. We have a special commitment to Disability & Inclusion, so we are interested in hiring people with disability certificate.