Cantina

Machine Learning Engineer, Images

Cantina
Apply
6 months ago
Remote, Worldwide or San Francisco, CA, USASenior
H1B Sponsor

Base Salary

$200k - $265k/yr

Responsibilities

  • Evaluate image-generation and identity-preservation research papers and models.
  • Develop and deploy new versions of image-generation and image-analysis pipelines.
  • Monitor and resolve production issues affecting users.
  • Fine-tune and optimize models for character consistency, prompt responsiveness, and inference latency.
  • Design and run experiments to benchmark model performance and track quality metrics across pipeline improvements.
  • Collaborate with cross-functional teams to translate product requirements into machine learning solutions and bring generative features from prototype to production.

Requirements

  • Demonstrated interest in AI image generation through personal or professional projects.
  • Deep technical foundation in machine learning, specifically image synthesis.
  • At least 5 years of software engineering experience, preferably building services.
  • At least 2 years of experience building production-grade machine learning models in industry or academic research settings.
  • Strong Python programming skills and experience deploying Python-based services.
  • Familiarity with Stable Diffusion, Diffusion Transformers, Visual Transformers, TensorFlow, PyTorch, Diffusers, ComfyUI, TensorRT, and CUDA.
  • Experience building end-to-end scalable ML infrastructure on on-premise or cloud platforms, including Baseten, Google Cloud Platform, Amazon Web Services, or Azure.
  • Strong communication, teamwork, and collaboration skills with technical and non-technical stakeholders.

Benefits

  • Generous company equity
  • Medical, dental, and vision insurance with 99.99% of premiums covered by Cantina
  • 42 days of paid time off, including PTO, sick days, company holidays, and floating holidays
  • Generous parental leave and fertility support
  • 401(k) retirement savings plan
  • $500 per month lifestyle spending account
  • Complimentary lunch and snacks for in-office employees
  • One Medical membership
Cantina

About Cantina

201-500 employees

Cantina Labs is a social AI company, developing a suite of advanced real-time models that push the boundaries of expression, personality, and realism. We bring characters to life, transforming how people tell stories, connect, and create. We build and power ecosystems. Cantina, our flagship social AI platform, is just the beginning.