🏢 Offshore Consulting Shops

Forte Group is a software development and IT consulting firm that provides staff augmentation and project-based services. The job description explicitly states that the role is for an external client ('Our Healthcare Tech client'), making this a client-facing position.

This company was flagged and excluded from default search results. Proceed with caution.

Forte Group
Posted 6mo ago

Lead/Principal Machine Learning Engineer (Generative Diffusion Models)

Forte Group
Mexico
RemoteFull Time
Responsibilities
  • Lead team
  • Develop architecture
  • Train diffusion models
Requirements
  • Lead role with expert PyTorch
  • Training diffusion models
  • Team leadership, and architectural ML work
Technical tools mentioned
PyTorchHuggingFace Diffusersdiffusion modelsU-NetsVAEs

Job description

Great software doesn’t happen on its own. It takes great people. That just happens to be our forte. With nearly 20 years of matching top engineering talent with preeminent and innovative brands, we look for people who are inquisitive, resourceful, dedicated to their craft, and driven to help companies build great software. If this sounds like you, read on.


You will join a highly technical team building a custom diffusion-based generative model and the surrounding training and data infrastructure. This is not fine-tuning an existing model; it is a ground-up implementation of a new architecture designed for high-end creative production.


Requirements

Must haves:

  • This role will be a Team Lead and requires prior leadership experience
  • Expert-level PyTorch
  • Proven experience training custom diffusion models
  • Deep understanding of attention mechanisms, U-Nets, VAEs, schedulers
  • Ability to implement modular model components from architectural specs
  • Training optimization (mixed precision, batching, checkpointing)
  • Clean coding practices and strong documentation
  • Ability to evaluate and improve visual model outputs

Nice to haves:

  • Experience with HuggingFace Diffusers
  • Work in image/video generation
  • Knowledge of control nets, inpainting, style conditioning

We offer

  • Work your way — anywhere, anytime. Our remote-first approach lets you choose where and how you work best!
  • Experience working with diverse teams and gaining international expertise
  • A friendly, supportive team and an enjoyable work environment where your ideas matter
  • A chance to work on exciting, challenging projects using cutting-edge technologies that make a real impact
  • Comprehensive health insurance, corporate psychologist access, and partial sports activity coverage
  • Free training programs, reimbursement for certifications, and access to online learning platforms to fuel your growth
  • Paid vacation, public holidays, and sick leave are fully covered by Forte Group
  • Referral bonuses, regular performance reviews, and full support for business trips
  • Corporate events and holiday presents


Join a team that invests in your well-being, growth, and success!


Similar jobs

Machine Learning Engineer roles
5d
Save
Mark Applied
Hide
Sr. Machine Learning Engineer
Monterrey or Nuevo Leon
$69k/yr HybridFull Time
Corning
CorningNYSE: GLW: Develops and manufactures glass, ceramics, and optical fiber products.
3+ YOEBachelor's or master's degree in a technical field, 3+ years of relevant experience, production ML deployment experience, Python, ML frameworks, MLOps, DevOps, CI/CD, and advanced technical and business English.
Python, scikit-learn, TensorFlow, PyTorch, Databricks, MLflow, Kubeflow, Docker, Kubernetes, APIs, CI/CD
1w
Save
Mark Applied
Hide
Machine Learning Engineer
Mexico City or State of Mexico or Mexico or South America
HybridFull Time
Konfío
Konfío: Financial platform providing credit and software to Mexican SMEs.
3+ YOEBachelor's in CS/Software/Math/Statistics, specialization in ML/Data Science, 3+ years production programming with Spark and Python, Docker/Kubernetes deployment, AWS and Google Cloud experience, Python and SQL proficiency.
Spark, Python, Docker, Kubernetes, AWS, Google Cloud, SQL, Big Data
2w
Save
Mark Applied
Hide
Machine Learning Engineer
Reynosa, Tamaulipas, Mexico
OnsiteFull Time
Corning
CorningNYSE: GLW: Develops and manufactures specialty glass, ceramics, and optical fiber.
3+ YOEBachelor's in CS, Software, Data Science, Industrial or Systems Engineering; 3+ years manufacturing/industrial experience; proven ML/AI automation delivery; strong programming, data analysis, project skills; advanced English.
OCR, RPA, Power BI, Tableau
3w
Save
Mark Applied
Hide
Lead Machine Learning Engineer
Mexico
RemoteFull Time
NobleAI
NobleAI: Science-based AI platform for industrial product R&D optimization.
5+ YOEMSc preferred; 5+ years building/deploying NLP/ML systems, strong Python experience, RAG and vector DB familiarity, cloud deployment (Azure), Docker/Kubernetes, excellent communication.
Python, PyTorch, TensorFlow, Retrieval-Augmented Generation (RAG), Pinecone, Weaviate, ChromaDB, LoRA, QLoRA, Llama, Mistral, OpenAI, Anthropic, Azure, Docker, Kubernetes, KFP, KServe, Git
3w
Save
Mark Applied
Hide
Ingeniero de Machine Learning / IA
Mexico City, Mexico City, Mexico
OnsiteContract
Sener
Sener: Engineering and technology group delivering global infrastructure and aerospace solutions.
1+ YOEBachelor's in Systems/Computer Science/Mathematics/Data Science, English B2,1–3 years ML/AI experience, Python, ML libraries (Scikit-learn, PyTorch, TensorFlow), data analysis (Pandas), experience with LLM APIs, Docker, and API development.
Python, Scikit-learn, PyTorch, TensorFlow, Pandas, OpenAI, Azure OpenAI, FastAPI, Flask, Django, Docker, PostgreSQL, SQLite, ChromaDB, Pinecone, PGVector
3w
Save
Mark Applied
Hide
Líder Técnico Machine Learning - Search Engine - Marketplace
Mexico or Argentina or Colombia
HybridFull Time
Mercado Libre
Mercado LibreNASDAQ: MELI: Leading e-commerce and fintech ecosystem in Latin America.
Design and operate production ML services with focus on scalability, resiliency, observability; experience with TensorFlow/PyTorch, Python and Java/Go, data pipelines, CI/CD, feature store, model serving, and technical leadership.
TensorFlow, PyTorch, Python, Java, Go, CI/CD
3w
Save
Mark Applied
Hide
Líder Técnico Machine Learning - Search Engine - Marketplace
Mexico or Argentina or Colombia
HybridFull Time
Mercado Libre
Mercado LibreNASDAQ: MELI: E-commerce and financial services marketplace for Latin America.
Design and operate production ML systems focused on scalability, resiliency, observability; strong Python and Java/Go; experience with TensorFlow or PyTorch, data pipelines, CI/CD, feature stores, model serving; technical leadership and communication skills.
TensorFlow, PyTorch, Python, Java, Go, CI/CD, feature store, experiment tracking, model serving, knowledge graphs, LLMs
3w
Save
Mark Applied
Hide
Oportunidades profesionales: CONSULTOR C2 INGENIERO ML (101841)
Mexico City, Mexico City, Mexico
HybridFull Time
Gentera
GenteraBolsa Mexicana de Valores: GENTERA: Provides financial products and services to underserved micro-entrepreneurs.
2+ YOEAt least 2 years of Python experience applied to machine learning, MLOps, CI/CD, and deploying, monitoring, or maintaining production models. Cloud, ML services, data pipelines, and data governance are a plus.
Python, CI/CD, MLOps