Lead AI Engineer (LLMs & Data Pipelines)

TLDR

Drive the design and operational excellence of LLM-powered capabilities across platforms, integrating models and optimizing performance in real-world systems.

Overview:

We are seeking a Lead AI Engineer (LLMs & Data Pipelines) to drive the design, integration, and operational excellence of LLM-powered capabilities across our platforms.

In this role, you will build intelligent features such as classification, extraction, summarization, and action orchestration powered by large language models. You will design embedding and retrieval pipelines (RAG, semantic search), create robust data pipelines for training and evaluation, and define clear evaluation metrics and quality gates to ensure reliable LLM behavior in production.

You will work hands-on with inference runtimes such as ONNX Runtime and TensorFlow Lite, benchmarking performance across CPU, GPU, NPU, and DSP environments, and optimizing deployments for latency, cost, and reliability—including in constrained or embedded systems. Collaborating with engineering, data, and MLOps teams, you will integrate models into real-world APIs and production systems while continuously experimenting with prompts, architectures, and model choices.

If you are passionate about turning advanced AI research into scalable, production-ready systems and efnjoy balancing performance, accuracy, and operational constraints, this may be your next mission.

What will you do?

  • Build and integrate LLM-powered features (classification, extraction, summarization, actions).
  • Integrate models with inference runtimes (such as ONNX Runtime, TensorFlow Lite / LiteRT).
  • Benchmark and validate model performance across different hardware backends (CPU, GPU, NPU, DSP).
  • Design embedding and retrieval pipelines (RAG, semantic search).
  • Create and maintain data pipelines for training and evaluation.
  • Define evaluation metrics and quality gates for LLM behavior.
  • Optimize inference for latency, cost, and reliability.
  • Integrate models into production systems and APIs.
  • Run experiments to evaluate prompts, models, and architectures.

What are we looking for?

  • Strong experience with LLMs and NLP systems
  • Hands-on experience with embeddings and vector databases
  • Strong Python skills and ML frameworks
  • Experience building production data pipelines
  • Solid understanding of evaluation and regression detection
  • Experience with RAG architectures
  • MLOps or monitoring experience 
  • Experience with model calibration and accuracy/latency trade-off analysis
  • Hands on experience deploying models on edge or embedded devices (constrained environments) 

What can you expect from us?

  • A permanent job contract for a long-term project;

  • Tech equipment + SIM Card + personal smartphone;

  • Health and Life Insurance;

  • Social events and team buildings;

  • The commitment of letting you grow with us, and be rewarded accordingly;

  • A dynamic and young team that will always be there to support you;

  • Training in the latest technologies;

  • Coffee, fruits, snacks, and a warm welcome whenever you visit the office.

Benefits

Free Meals & Snacks

Coffee, fruits, snacks, and a warm welcome whenever you visit the office

Health Insurance

Health and Life Insurance

Training opportunities

Training in the latest technologies

Caixa Mágica Software develops advanced software solutions tailored for the automotive industry, including embedded systems and Android Automotive applications, enabling seamless in-vehicle experiences. Targeting both consumers and businesses, they modernize existing SAP systems while providing a diverse range of IT and business consulting services.

View all jobs
Ace your job interview

Understand the required skills and qualifications, anticipate the questions you may be asked, and study well-prepared answers using our sample responses.

AI Engineer Q&A's
Report this job
Apply for this job