Machine Learning / AI Engineer (Hybrid Madrid)

Skills

Location

Languages

Fluent English

Job Description

We're looking for a Machine Learning / AI Engineer to collaborate with a major client in the banking sector.

🎯 Responsibilities

  1. Develop, deploy, and optimize machine learning and AI solutions for complex business challenges
  2. Create multi-agent systems and equip AI models with function/tool-calling capabilities
  3. Design and implement Retrieval-Augmented Generation (RAG) systems to ground AI responses in enterprise data
  4. Assess and integrate AI models (such as LLMs) to ensure optimal performance and reliability
  5. Optimize agentic workflows and AI agents for production use cases
  6. Test and optimize system prompts and few-shot examples to ensure accurate, consistent, and safe AI outputs
  7. Evaluate the performance of machine learning and AI models using appropriate metrics, evaluation sets, and techniques, with ongoing improvements
  8. Collaborate with platform teams, data engineers, data scientists, and other stakeholders to integrate machine learning solutions into existing systems and processes
  9. Participate in code reviews, testing, and debugging to ensure quality and reliability

🛠️ Requirements

  1. Bachelor's or Master's degree in Data Science, Computer Science, Mathematics, Statistics, or related field
  2. Strong programming skills in Python, with experience in machine learning libraries and deep learning frameworks such as TensorFlow or PyTorch
  3. Proven experience working with Large Language Models (LLMs)
  4. Good understanding of AI agents, agentic workflows, LLM orchestration frameworks, and reasoning patterns
  5. Experience with data preprocessing, feature engineering, and model selection and evaluation techniques
  6. Knowledge of statistical and mathematical concepts relevant to machine learning, including probability, linear algebra, and optimization
  7. Understanding of software development best practices, including version control, testing, and documentation
  8. Excellent problem-solving and debugging skills
  9. Relevant professional experience in machine learning, data science, or related field
  10. Good English level (C1), as you will be working with international teams

🤗 We offer

  1. 23 days of annual leave, with the 24th and 31st of December as additional discretionary days
  2. Comprehensive benefits (healthcare plan, teleworking compensation, life and accident insurance)
  3. Flexible compensation program, including meals, kindergarten, transport, online English lessons, and healthcare plan
  4. Free access to multiple training platforms
  5. Professional stability and clear career plans
  6. Referral bonuses for recommending professionals
  7. Option to select payment structure (12 or 14 payments per year)
  8. Work-life balance measures, such as flexibility, WFH/remote work policy, and compacted hours during summertime
  9. Discounts through a club platform and on gym memberships

🕒 Work Schedule

  1. Business hours (standard working days; no special intensive schedules for Fridays or summer)

📍 Work Location

  1. Hybrid model: 3 days per week in the office (Madrid city center, Sol area), 2 days remote

🔑 Skills

  1. Machine learning
  2. Python
  3. RAG architecture
  4. PyTorch
Each company upload its own logo image

UST

UST is an IT services company and IT solutions provider with a global presence. Headquartered in California, for over 20 years, we have worked side by side with the world’s best companies to make a real impact through transformation. Powered by technology, inspired by people and led by our purpose, we partner with industry leaders from design to operation. With over 26,000 employees in 25 countries, 5 continents, we build for boundless impact—touching billions of lives in the process.

Benefits and perks

  • workFromHome

    Work from home

  • flexible

    Flexible schedule

  • health

    Health insurance

  • gym

    Gym discount

  • +23 days of Annual Leave plus the 24th and 31st of December as discretionary days
  • +Internet Connectivity, Life and Accident Insurances
  • +`Retribución Flexible´ Program: you will benefit from a flexible salary package that allows to optimise your income

Location

Calle Santa Leonor, 65, Madrid, Spain