Machine Learning Engineer, Images

Posted 25ds ago

Employment Information

Education
Salary
Experience
Job Type

Report this job

Job expired or something wrong with this job?

Job Description

Senior Machine Learning Engineer developing AI image generation models for a social AI company focused on enhancing creativity and connections. Join a team revolutionizing online interaction through AI.

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