Background

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:

Essential Skills:

• Advanced expertise with deep learning libraries such as PyTorch, Jax, or TensorFlow, demonstrated through substantial professional experience
• Proven track record in high-impact software or model engineering, with significant contributions evidenced by publications, patents, or advanced degrees
• Mastery of data pipeline architecture and optimization for large-scale machine learning systems
• Extensive 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 domain
• Strong programming skills in C/C++

Our Culture

At Udio, we are driven by excellence and innovation. We value:

- A deep passion for creating transformative products
- An aggressive approach to feature deployment and rapid iteration
- A genuine love for music and the art of creation
- An enthusiasm for generative AI and pioneering technology
- An unwavering commitment to excellence and integrity

Benefits

- Highly competitive salary and equity compensation
- Flexible time off (~25 days PTO/year)
- Prime office locations in Manhattan (NYC) or Shoreditch (London)
- Productivity tools including ChatGPT Plus and Copilot
- Comprehensive private health, dental, and vision insurance for you and your dependents
- 401(k) plan with 5% employer matching
- Concierge medical/primary care through One Medical and Rightway
- Robust mental health support via Spring Health
- Personalized life insurance, travel assistance, and additional high-caliber perks

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