Senior AI ML Engineer - Databricks - #2
🚀 Join Our Remote Data Products & Machine Learning Startup! 🚀
At Muttdata, we build innovative Data Products and Machine Learning solutions that help companies solve complex business challenges. As a fast-growing, remote-first startup, we're passionate about technology, collaboration, and continuous learning.
This opportunity is with a leading multinational beverage company based in Mexico City.
We are looking for an innovative Senior AI ML Engineer to join our team 🐶🚀. You'll build an AI agent for the Model Factory Lab (MFL) community that turns a natural-language prompt into a fully instantiated model — orchestrating reusable catalog components under the “model as a YAML” paradigm.
This role works closely with the MFL platform team, combining agent development with a deep understanding of model packaging and reusable, governed components. Strong technical creativity and a product mindset are essential to succeed in this fast-paced, collaborative environment.
🚀 What We Do
- Leveraging our expertise, we build modern Machine Learning systems for demand planning and budget forecasting.
- Developing scalable data infrastructures, we enhance high-level decision-making, tailored to each client.
- Offering comprehensive Data Engineering and custom AI solutions, we optimize cloud-based systems.
- Using Generative AI, we help e-commerce platforms and retailers create higher-quality ads, faster.
- Building deep learning models, we enhance visual recognition and automation for various industries, improving product categorization, quality control, and information retrieval.
- Developing recommendation models, we personalize user experiences in e-commerce, streaming, and digital platforms, driving engagement and conversions.
🌟 Our Partnerships
- Amazon Web Services
- Astronomer
- Databricks
🌟 Our Values
- 📊 We are Data Nerds
- 🤗 We are Open Team Players
- 🚀 We Take Ownership
- 🌟 We Have a Positive Mindset
🔍 Curious about what we’re up to? Check out our case studies and dive into our blog post to learn more about our culture and the exciting projects we’re working on! 🚀
Responsibilities 🤓
- Design and build the agent that translates natural language into a composition of trainable components (train/predict/evaluate), following the MFL development framework (PyFunc / custom flavor as standard).
- Implement the model's declarative configuration in YAML (config.yaml, features.yaml, databricks.yml), respecting the MFL's minimum library skeleton.
- Integrate the agent with Feature Store, MLflow, and model orchestration to assemble governed, reusable components.
- Apply production-grade agentic patterns on Databricks (Mosaic AI Agent Framework, Agent Bricks, MCP Servers, evaluation with MLflow 3.0).
Required Skills 💻
- Experience developing AI agents, ideally on Databricks (Mosaic AI, LangGraph / OpenAI SDK, Genie, Vector Search).
- Strong command of MLflow (tracking, registry, pyfunc / custom flavor), Unity Catalog, and advanced Python.
- Understanding of component-orchestration architectures and model packaging.
- Experience with MCP, RAG, agent evaluation, and model governance standards.
- Broader experience developing AI agents / agentic infrastructure (Mosaic AI Agent Framework, agent orchestration, MCP).
🎁 Perks
- Remote-first culture – work from anywhere! 🌍
- AWS, DBT, Google Cloud, Azure & Databricks certifications fully covered
- Birthday off + an extra vacation week (Mutt Week! 🏖️)
- Referral bonuses – help us grow the team & get rewarded!
- Maslow: Monthly credits to spend in our benefits marketplace.
- ✈️🏝️ Annual Mutters' Trip – an unforgettable getaway with the team!