Cantina

Building the first social AI platform

Machine Learning Engineer, Images

Full TimeRemoteTeam 51-200Since founded by Sean ParkerH1B SponsorCompany SiteLinkedIn

Location

California

Posted

20 hours ago

Salary

$200K - $265K / year

Bachelor Degree5 yrs expEnglishAWSAzureCloudGoogle Cloud PlatformPythonPy TorchTensorflow

Job Description

• 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

Job 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!

Related Job Pages