AI Platform Engineer / AI Infrastructure Engineer

Axiomatic_AI; Barcelona; Hybrid; EUR 70,000 - 90,000; Full time

Benefits offered by this job

Competitive compensation

Stock Options

Flexible work arrangements

Opportunities for professional growth

Job summary

A leading AI technology firm in Barcelona is seeking an AI Platform Engineer to build and maintain AI infrastructure, enabling developers to deploy work safely. The candidate will work on API design, implement MLOps pipelines, and optimize performance. Required skills include 7+ years of software engineering experience, proficiency in Python, and experience with LLMs in production. This role offers competitive compensation, flexible work arrangements, and opportunities for professional growth.

Qualifications

• 7+ years of software engineering experience, Python preferred.
• Experience with LLMs and AI/ML in production: OpenAI API, HuggingFace, LangChain.
• Knowledge of vector databases: Pinecone, Chroma, Weaviate.

Responsibilities

• Build and maintain AI infrastructure: model serving, vector databases.
• Enable AI developers to deploy reproducibly and safely.
• Design APIs for AI inference and evaluation.

Skills

Software engineering experience

Experience with LLMs and AI/ML in production

Knowledge of vector databases

Cloud infrastructure experience

API development

CI/CD and DevOps

Monitoring and observability

Problem-solving mindset

Experience deploying AI at enterprise level

Tools

Python

Docker

FastAPI

Terraform

GitHub Actions

Job description

Join to apply for the AI Platform Engineer / AI Infrastructure Engineer role at Axiomatic_AI

About us

Axiomatic_AI is dedicated to accelerating R&D by developing the next generation of Automated Interpretable Reasoning, a verifiably truthful AI model built for reasoning in science and engineering. We empower engineers in hardware design and Electronic Design Automation (EDA), with a mission to revolutionize the fields of hardware design and simulation in the photonics and semiconductor industry.

Position Overview

As an AI Platform Engineer, you will bridge AI research and production software, building and maintaining AI infrastructure, enabling developers to deploy work safely, and designing APIs for AI inference, prompt management, and evaluation. You will also implement MLOps pipelines, optimize performance, and collaborate closely with backend engineers.

Key Responsibilities

• Build and maintain AI infrastructure: model serving, vector databases, embedding pipelines.
• Enable AI developers to deploy reproducibly and safely.
• Design APIs for AI inference, prompt management, and evaluation.
• Implement MLOps pipelines: versioning, monitoring, logging, experimentation tracking.
• Optimize performance: latency, cost, throughput, and reliability.
• Collaborate with backend engineers to integrate AI capabilities.
• Monitor model performance, drift, and set up logging and observability.
• Build CI/CD pipelines for model deployment.
• Document AI infrastructure and best practices.
• Mentor AI developers on software practices.

Required Skills & Experience

• 7+ years of software engineering experience (Python preferred).
• Experience with LLMs and AI/ML in production: OpenAI API, HuggingFace, LangChain, or similar.
• Knowledge of vector databases: Pinecone, Chroma, Weaviate, FAISS.
• Cloud infrastructure experience: GCP (Vertex AI preferred) or AWS (SageMaker).
• API development: FastAPI, REST, async programming.
• CI/CD and DevOps: Docker, Terraform, GitHub Actions.
• Monitoring and observability.
• Problem‑solving mindset; comfortable debugging complex distributed systems.
• Experience deploying AI at enterprise level.

Nice-to-Have

• Fine‑tuning or training models.
• Familiarity with LangChain, Pydantic AI, or similar frameworks.
• Prompt engineering and evaluation techniques.
• Real‑time inference and streaming responses.
• Background in data engineering or ML engineering.
• Knowledge of RAG architectures.
• Contributions to open‑source AI/ML projects.

Tech Stack

• Languages: Python, Bash.
• AI/ML: OpenAI API, Anthropic, HuggingFace, LangChain, Pydantic AI.
• Vector DBs: Pinecone, Chroma, Weaviate, FAISS.
• Backend: FastAPI, SQLAlchemy, Pydantic.
• Cloud: GCP (Vertex AI, Cloud Run), Terraform.
• CI/CD: GitHub Actions.
• Experiment Tracking: MLflow, Weights & Biases, or custom.
• Containers: Docker; Kubernetes optional.

What We Offer

• Competitive compensation, including Stock Options.
• Access to state‑of‑the‑art tools and collaboration with leading experts.
• Flexible work arrangements with potential remote options.
• Opportunities for professional growth: conferences, conferences, publications.
• A culture focused on impact in AI and hardware.

Why Join Us

At Axiomatic_AI we drive innovation in AI for scientific and engineering applications. Your contributions will help bring new AI architectures that reason coherently and produce verifiable solutions to market, shaping the future of hardware and computing. We celebrate diversity and encourage applicants who may not meet every qualification to apply; we are eager to consider a wide range of experiences and backgrounds.