Background

At Udio, our mission is to enable the next generation of music creators with state-of-the-art AI tools. As a machine learning engineer, you will help build and fine-tune the foundation models which power Udio. We are hiring both senior and junior machine learning engineers.

Our founding team is composed of world-leading AI researchers, formerly of Google DeepMind. Between them, our founders have 4 PhDs, degrees from Harvard, Oxford, Mila, and Edinburgh, and industry experience at DeepMind, Google, Nvidia, Dropbox, Yext, and more.

We are backed by Andreesen Horowitz, along with leading lights from tech, music, and creative arts worlds, including will.i.am, Common, Tay Keith, United Masters, and Kevin Wall, Mike Krieger (co-founder of Instagram), and Oriol Vinyals (head of Google Gemini).

Core Technical Competencies

The following competencies suggest you’ll be a good candidate for this role. You don’t need to meet all the criteria below, but the more you meet the better!

Must Haves

Top class software-or-model-engineering abilities as demonstrated by experience, publications, and/or education
One of:
- Familiarity with a deep learning library: Pytorch, Jax, or Tensorflow
- Data pipeline expertise
- Large scale model training experience: e.g. debugging and optimizing TPU or GPU models, model parallelism, kernel languages (CUDA/Triton/Pallas)

Nice to Haves

Experience fine-tuning open source models, especially LLMs
Familiarity with training on GKE
Generative modeling expertise (especially with audio)
Knowledge of C/C++

Our Culture

We love doing things exceptionally well, and we have fun doing so! Some cultural traits we value are:

Passion for creating products people love
A bias towards action: shipping features to consumers at a rapid pace
Love of music
Deep interest in generative AI and pioneering technology
Commitment to excellence and integrity

Benefits

Highly competitive salary and equity
Flexible time off (~25 days PTO/year)
Fantastic office location in Manhattan (NYC) or Shoreditch (London)
Productivity package, including ChatGPT Plus and Copilot
Top notch private health, dental, and vision insurance for you and your dependents
401(k) plan options with 5% employer matching
Concierge medical/primary care through One Medical and Rightway
Mental health support from Spring Health
Personalized life insurance, travel assistance, and many other perks

Udio’s success hinges on hiring great people and creating an environment where we can be happy, feel challenged, and do our best work.

Udio provides equal employment opportunities (EEO) to all employees and applicants for employment without regard to race, color, religion, sex, national origin, age, disability, genetics, sexual orientation, gender identity, or gender expression. We are committed to a diverse and inclusive workforce and welcome people from all backgrounds, experiences, perspectives, and abilities.

This role is eligible for a compensation package of base salary, equity, and benefits. The starting base salary range for this role is $150,000 - $300,000. Actual salary may vary based on level, work experience, performance, and other factors evaluated during the hiring process.

Udio jobs

Machine Learning Engineer