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

Requirements

Demonstrated interest in AI image generation. This includes both 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 of 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

Benefits

Competitive salary and generous company equity
Medical, dental, and vision insurance – 99.99% of premiums covered by Cantina
42 days of paid time off, including:
15 PTO days
10 sick days
15 company holidays
2 floating holidays
Generous parental leave & fertility support
401(k) retirement savings plan
Lifestyle spending account – $500/month to use however you’d like
Complimentary lunch and snacks for in-office employees
One Medical membership, and more!

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learningimage synthesisPythonproduction-grade machine learning modelsmodel optimizationbenchmarkingimage generationimage analysisinference latencyquality metrics

Soft Skills

teamworkcommunicationcollaboration