Senior Machine Learning Engineer - Databricks - #2

MuttdataMexicoOn-siteFull-timeSenior, 5–8 yearsListed 2 days ago

Apply now

About this role

πŸš€ 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 (https://blog.muttdata.ai/tag/Case%20Studies/1) case studies and dive into ourblog post (https://blog.muttdata.ai/) to learn more about our culture and the exciting projects we’re working on! πŸš€

Responsibilities πŸ€“

- Orchestrate production pipelines on Databricks Jobs, chaining tasks via depends_on and managing failure modes (ALL_SUCCESS / ALL_DONE)

- Configure and maintain deployments via Databricks Asset Bundles (DABs), promoting across dev β†’ qa β†’ prd targets, with resource/deployment files per country.

- Manage the lifecycle of models and artifacts in MLflow + Unity Catalog (registration, versioning, Champion/Challenger aliases, rollback).

- Ensure operational continuity: manage compute across workspaces, standardize cost-monitoring tags (FinOps), and integrate with AIOps/observability.

- Master the product end-to-end to diagnose runtime incidents and coordinate with development squads.

Required Skills πŸš€

- Proven experience orchestrating data/ML pipelines on Databricks (Jobs, Workflows, cluster policies).

- Solid CI/CD experience (Azure DevOps, gitflow, PRs), IaC with DABs, and code quality standards (black, isort, mypy).

- Advanced Python, SQL, Delta / Unity Catalog.

Nice to Have πŸ˜‰

- Experience with multi-country schemas, drift/cost monitoring, and deployment automation.

- Experience developing AI agents / agentic infrastructure (e.g. Mosaic AI Agent Framework, agent orchestration, MCP).

- Currently based in Mexico or a location within a similar time zone, allowing for strong overlap with Mexico working hours.

🎁 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 (https://maslow.hr/): Monthly credits to spend in our benefits marketplace.

- ✈️ 🏝️Annual Mutters' Trip (https://www.youtube.com/watch?v=dq1x2krHGb8&t=2s) – an unforgettable getaway with the team!