Lead AI Engineer

Salesforce · Enterprise · Mexico City, Mexico

Lead AI Engineer to drive the development of next-generation AI and ML systems, focusing on building an agent flywheel with self-improving feedback loops. The role involves designing and implementing feedback loops for agent evaluation and optimization, developing production ML models and AI agents that combine LLM reasoning and tool usage, and engineering data pipelines for training and evaluation. Emphasis on closing the loop from production signals to model improvements and building hybrid systems blending deterministic logic, model-based scoring, and LLM generation.

What you'd actually do

  1. Design and implement feedback loops that enable agents and ML models to self-improve over time
  2. Build and deploy application-specific ML models (classification, ranking, forecasting, recommendation, etc.)
  3. Design and implement AI agents that combine LLM reasoning, Tool/API usage, ML-based decisioning layers
  4. Design and build scalable data pipelines (batch and near real-time) that power training, evaluation, and inference workflows
  5. Build offline and online evaluation frameworks for agent and ML model performance

Skills

Required

  • Python for production systems
  • building and deploying production-grade ML models
  • data pipeline development (ETL/ELT, batch or streaming)
  • designing and building AI agents or agent-like systems
  • API development and backend services
  • ML lifecycle tooling (training, evaluation, deployment, monitoring)
  • building reliable data pipelines that support ML or AI systems in production

Nice to have

  • Data processing frameworks (e.g., Spark or equivalent)
  • Data orchestration tools (e.g., Airflow, Dagster, etc.)
  • Data warehousing solutions (e.g., Snowflake, BigQuery, etc.)
  • Understanding of data quality, lineage, and reproducibility in ML systems

What the JD emphasized

  • production-grade ML models
  • AI agents
  • agent workflows
  • ML models
  • evaluation frameworks
  • agent performance
  • ML systems
  • agent behavior
  • ML model performance
  • agent capabilities

Other signals

  • building agent flywheel
  • self-improving feedback loops
  • production-grade ML models
  • AI agents
  • continuous improvement in production