AI Engineer - Remote (Spain)

Lifebit is a pioneering company focused on harnessing connected data for precision medicine. As an AI Engineer, you will develop and deploy machine learning models that empower researchers to analyze decentralized biomedical datasets, transforming insights into actionable outcomes for drug development and disease research.

Responsibilities:

• Design and implement autonomous AI agents using frameworks like LangGraph, CrewAI, or AutoGen to handle complex, multi-step scientific queries
• Develop sophisticated reasoning loops (e.g., ReAct, Plan-and-Execute) that allow agents to decompose high-level research goals into actionable sub-tasks
• Build and optimize Advanced RAG (Retrieval-Augmented Generation) pipelines that integrate structured clinical data and unstructured scientific literature
• Create and maintain "tools" for AI agents, enabling them to safely interface with Lifebit’s federated APIs, SQL databases, and bioinformatic execution engines
• Implement secure, sandboxed code-interpreter capabilities, allowing agents to write and execute Python or R code for data visualization and statistical analysis
• Fine-tune LLMs for specific function-calling and tool-use accuracy within the life sciences domain
• Develop robust evaluation frameworks (LLM-as-a-judge) to measure agentic performance, truthfulness, and safety in a clinical context
• Implement "Human-in-the-loop" (HITL) patterns to ensure high-stakes scientific decisions are always reviewed by domain experts
• Partner with Security teams to ensure agents operate within strict data privacy boundaries, preventing prompt injection or unauthorized data egress in federated nodes
• Work with Product and UX teams to design intuitive interfaces for interacting with agentic systems (e.g., conversational research assistants)
• Scale agentic workloads in production using Kubernetes, ensuring low-latency reasoning and efficient token usage

Requirements:

• Education: BSc/MSc in Computer Science, Artificial Intelligence, Machine Learning, or a highly quantitative field (PhD preferred)
• Experience: 2+ years of hands-on experience as an AI or ML Engineer, building a validated real product, ideally within a product-led biotech, health-tech, or SaaS company
• Technical Stack: Deep proficiency in Python and Typescript and standard ML frameworks (e.g., Langfuse, PyTorch, TensorFlow, JAX, Scikit-learn)
• NLP/LLM Expertise: Proven experience working with Large Language Models, including fine-tuning, and RAG (Retrieval-Augmented Generation) architectures
• Cloud & Infrastructure: Familiarity with AWS/Azure/GCP and experience deploying models in Docker/Kubernetes environments
• Autonomy: A self-starter mindset with the ability to navigate ambiguity and drive AI projects from concept to production without constant oversight
• Domain Knowledge: It is a plus to have experience working with biological, genomic, or clinical data is a significant advantage