Logo
Cantina Consulting

Senior Machine Learning Engineer - Images

Cantina Consulting, San Francisco, California


A bit about Cantina: Cantina, founded by Sean Parker, is a new social platform with the most advanced AI character creator. Build, share, and interact with AI bots and your friends directly in the Cantina or across the internet. Cantina bots are lifelike, social creatures, capable of interacting wherever humans go on the internet. Recreate yourself using powerful AI, imagine someone new, or choose from thousands of existing characters. Bots are a new media type that offer a way for creators to share infinitely scalable and personalized content experiences combined with seamless group chat across voice, video, and text. If you're excited about the potential AI has to shape human creativity and social interactions, join us in building the future A bit about the role: As a Senior ML Engineer on the AI Image Generation (Imagine) team, you'll design, implement, fine tune, improve and debug the image and video AI models that power our lifelike AI bots. Your expertise in machine learning and scalable ML infrastructure will be crucial in developing innovative features that revolutionize how people connect and create online. The Imagine Team at Cantina is responsible for all generative image, video, and machine vision services. AI bots on Cantina are multimodal and can text and talk with you as well as send you selfies. To provide these capabilities we have developed a stable diffusion based image generation pipeline that creates photo realistic, consistent characters. The Imagine team is constantly striving to improve the quality, character consistency, responsiveness to prompting, and incorporation of an ever increasing number of custom looks and appearance traits. A bit about you: Demonstrated interest in AI image generation. This includes both personal and professional projects. Deep technical foundation in machine learning specifically in image synthesis 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, Visual Transformers (ViT), Tensorflow, PyTorch, Diffusers, ComfyUI, Segment Anything (SAM), Kohya-ss, CodeFormer, and GFPGAN. Experience building end-to-end scalable ML infrastructure with on-premise or cloud platforms including 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 Location: We have offices located in Sunnyvale, CA, San Francisco, CA and Brooklyn, NY. While we have a strong focus on individuals near our office hubs, we offer fully remote and hybrid employment opportunities. Pay Equity: In compliance with Pay Transparency Laws, the base salary range for this role is between $175,000-250,000 for those located in the San Francisco Bay Area, New York City and Seattle, WA. When determining compensation, a number of factors will be considered, including skills, experience, job scope, location, and competitive compensation market data. Benefits Summary: Health Care — 99% of premiums for medical, vision, dental are fully paid for by Cantina, plus One Medical membership. Monthly Stipend — $500/month to use on whatever you’d like Rest and Recharge — 15 PTO days per year, 9 sick days, 13 paid company holidays, and offices closed for winter break (Christmas Eve to New Years Day) 401(K) — Eligible to participate on day one of employment. Parental Leave & Fertility Support Competitive Salary & Equity Lunch and snacks provided for in-office employees. WFH equipment provided for full-time hybrid/remote employees. J-18808-Ljbffr