# Machine Learning Engineer, Images

**Company:** [Cantina](https://hotfix.jobs/companies/cantina)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $200k – $265k/yr
**Experience:** 5+ years
**Skills:** PyTorch, TensorFlow, Stable Diffusion, Diffusers, Kubernetes, Python, CUDA, TensorRT, GCP, AWS
**Posted:** 2026-03-05

> Designs, fine-tunes, and deploys image generation models for photorealistic AI bots, optimizing for consistency, latency, and quality. Requires 5+ years software engineering, 2+ years production ML, and expertise in diffusion models like Stable Diffusion and PyTorch.

## Job Description

## What You’ll Do
- Evaluate new image generation and identity preservation papers and models
- Develop and deploy new versions of the image generation and image analysis pipelines
- Monitor and fix production issues that impact users
- Fine-tune and optimize models to improve character consistency, prompt responsiveness, and inference latency
- Design and run experiments to benchmark model performance, tracking quality metrics across generations of pipeline improvements
- Collaborate with cross-functional teams to translate product requirements into ML solutions and bring new generative features from prototype to production

## What You’ll Bring
- Demonstrated interest in AI image generation (personal and professional projects)
- Deep technical foundation in machine learning specifically in image synthesis
- 5+ years experience as a software engineer, preferably in services
- 2+ years of experience building production-grade machine learning models in industry and/or academic research settings
- Strong programming skills in Python and deploying Python based services
- Familiarity with tools and frameworks involved in AI image generation including but not limited to **Stable Diffusion**, **Diffusion Transformers (DiT)**, **Visual Transformers (ViT)**, **Tensorflow**, **PyTorch**, **Diffusers**, **ComfyUI**, **TensorRT**, and **CUDA**
- Experience building end-to-end scalable ML infrastructure with on-premise or cloud platforms including **Baseten**, **Google Cloud Platform (GCP)**, **Amazon Web Services (AWS)** or **Azure**
- Strong teamwork skills including communication and collaboration with both technical and non-technical team members

## Compensation
- Anticipated annual base salary range: **$200,000-$265,000**


## Similar roles

- [Research Engineer](https://hotfix.jobs/jobs/b4d18018-e64d-4828-a8b4-303225328e17) - Console - San Francisco, CA - $200k – $350k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/1b64040d-77ad-441e-85bb-9de3583cf9b8) - Kepler - New York, NY - $200k – $280k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/bc31e7f5-d976-4a97-a1be-bee1a0b92f94) - AfterQuery - San Francisco, CA - $200k – $300k/yr
- [Machine-Learning Operations Engineer](https://hotfix.jobs/jobs/d80db2a3-5a47-41f4-b4ae-220a73b77cbf) - Tennr - New York, NY - $200k – $230k/yr
- [Machine Learning Engineer, Enterprise Brain](https://hotfix.jobs/jobs/b5ce4e0d-3ddb-4ef6-82aa-18b21d4639ff) - Glean - Mountain View, CA - $200k – $300k/yr

**Apply:** https://hotfix.jobs/jobs/40fe74dd-cb19-4510-8154-93efd97cc81f
**Canonical:** https://hotfix.jobs/jobs/40fe74dd-cb19-4510-8154-93efd97cc81f