Senior ML Engineer - AI Image Generation & Vision

Cantina Labs; San Francisco (CA); Remote; USD 200,000 - 265,000; Full time

Benefits offered by this job

Competitive salary

Generous company equity

Medical, dental, and vision insurance

42 days of paid time off

Generous parental leave & fertility support

401(k) retirement savings plan

Lifestyle spending account

Complimentary lunch and snacks

Job summary

An innovative social AI company in San Francisco is seeking a Senior Machine Learning Engineer to design and enhance image AI models. This role involves developing scalable ML infrastructure and optimizing models for character consistency and responsiveness. Candidates should have over 5 years of engineering experience and expertise in AI image generation tools such as TensorFlow and PyTorch. The position offers a competitive salary, substantial paid time off, and various benefits including medical insurance and retirement plans.

Qualifications

• 5+ years of experience as a software engineer, preferably in services.
• 2+ years of experience building production‑grade machine learning models.
• Strong programming skills in Python and deploying Python based services.

Responsibilities

• 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.

Skills

AI image generation

Machine learning

Python programming

Collaboration

Tools

Stable Diffusion

Tensorflow

PyTorch

CUDA

Job description

An innovative social AI company in San Francisco is seeking a Senior Machine Learning Engineer to design and enhance image AI models. This role involves developing scalable ML infrastructure and optimizing models for character consistency and responsiveness. Candidates should have over 5 years of engineering experience and expertise in AI image generation tools such as TensorFlow and PyTorch. The position offers a competitive salary, substantial paid time off, and various benefits including medical insurance and retirement plans.