Logo
Tbwa Chiat/Day Inc

Staff Software Engineer, Machine Learning

Tbwa Chiat/Day Inc, New York, New York, us, 10261


Staff Software Engineer, Machine Learning

New York City, New York, United StatesBackground

At Udio, we are at the forefront of revolutionizing music creation with cutting-edge AI tools. As a seasoned machine learning engineer, you will play a pivotal role in architecting and refining the foundational models that power our innovative platform.Our founding team comprises leading AI researchers from Google DeepMind, each bringing unparalleled expertise. With 4 PhDs and degrees from prestigious institutions such as Harvard, Oxford, Mila, and Edinburgh, their backgrounds include significant tenures at top tech giants including DeepMind, Google, Nvidia, Dropbox, and Yext.We are backed by Andreessen Horowitz, alongside influential figures from tech, music, and the creative arts, including will.i.am, Common, Tay Keith, United Masters, as well as Kevin Wall, Mike Krieger (co-founder of Instagram), and Oriol Vinyals (head of Google Gemini).Core Technical Competencies

As a Staff Level Machine Learning Engineer, you are expected to excel in the following areas. While not all criteria are mandatory, alignment with these competencies is highly valued:Advanced expertise with deep learning libraries such as PyTorch, Jax, or TensorFlow, demonstrated through substantial professional experienceProven track record in high-impact software or model engineering, with significant contributions evidenced by publications, patents, or advanced degreesMastery of data pipeline architecture and optimization for large-scale machine learning systemsExtensive experience with large-scale model training, including debugging and optimizing TPU/GPU models, model parallelism, and kernel languages (e.g., CUDA, Triton, Pallas)Deep expertise in fine-tuning open-source models, particularly large language models (LLMs)Advanced knowledge of training and deploying models on Google Kubernetes Engine (GKE)Proficiency in generative modeling, especially in the audio domainStrong programming skills in C/C++Our Culture

At Udio, we are driven by excellence and innovation. We value:A deep passion for creating transformative productsAn aggressive approach to feature deployment and rapid iterationA genuine love for music and the art of creationAn enthusiasm for generative AI and pioneering technologyAn unwavering commitment to excellence and integrityBenefits

We offer:Highly competitive salary and equity compensationFlexible time off (~25 days PTO/year)Prime office locations in Manhattan (NYC) or Shoreditch (London)Productivity tools including ChatGPT Plus and CopilotComprehensive private health, dental, and vision insurance for you and your dependents401(k) plan with 5% employer matchingConcierge medical/primary care through One Medical and RightwayRobust mental health support via Spring HealthPersonalized life insurance, travel assistance, and additional high-caliber perksAt Udio, our success hinges on attracting top-tier talent and fostering a high-performance environment where you can excel, innovate, and drive impactful results.Udio is an equal opportunity employer and is committed to creating a diverse and inclusive workplace. We encourage applications from individuals of all backgrounds and experiences.The compensation package for this role includes a competitive base salary, equity, and benefits.

The starting base salary range is $150,000 - $300,000, with variations based on experience, achievements, and other factors evaluated during the hiring process.Apply for this job

To apply, please submit the following information:First Name *Last Name *Email *Phone *Resume/CV *LinkedIn Profile *WebsitePlease list and explain your experience with Large scale model training *Describe a deep learning modeling project (not necessarily generative AI) you’ve worked on. What was your role? *Please list and explain your experience with large-scale data pipelines (ingestion, filtering, formatting etc) *Please list and explain your experience with large-scale model inference (deployment, versioning, optimization etc) *Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.As set forth in Udio’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

#J-18808-Ljbffr