Machine Learning Engineer

Job expired!

Join Udio as a Machine Learning Engineer

At Udio, we empower the next generation of music creators with cutting-edge AI tools. We are seeking both senior and junior Machine Learning Engineers to join our team and help build and optimize the foundational models driving Udio.

About Udio

Our founding team comprises world-leading AI researchers with extensive experience from Google DeepMind. Collectively, our founders hold four PhDs and degrees from prestigious institutions such as Harvard, Oxford, Mila, and Edinburgh. They have worked with renowned companies like DeepMind, Google, Nvidia, Dropbox, Yext, and more.

Udio is backed by Andreesen Horowitz and influential figures from the tech, music, and creative arts industries, including will.i.am, Common, Tay Keith, United Masters, Kevin Wall, Mike Krieger (co-founder of Instagram), and Oriol Vinyals (head of Google Gemini).

Core Technical Competencies

The following competencies indicate a strong fit for this role. While you don’t need to meet all criteria, the more you align with, the better.

Must Haves

  • Exceptional software or model engineering skills demonstrated through experience, publications, or education
  • Proficiency in one of the following:
    • Deep learning libraries such as Pytorch, Jax, or Tensorflow
    • Data pipeline expertise
    • Experience with large scale model training, including debugging and optimizing TPU or GPU models, model parallelism, and kernel languages like CUDA/Triton/Pallas

Nice to Haves

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

Our Culture

At Udio, we take pride in our work and enjoy the process! Key cultural traits we value include:

  • Passion for creating products that people love
  • A bias towards action, with a focus on rapidly shipping features to consumers
  • Love of music
  • Deep interest in generative AI and pioneering technology
  • Commitment to excellence and integrity

Benefits

  • Highly competitive salary and equity compensation
  • Approximately 25 days of PTO per year
  • Fantastic office locations in Manhattan (NYC) or Shoreditch (London)
  • Productivity package including ChatGPT Plus and Copilot
  • Top-tier 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

At Udio, our success relies on hiring exceptional people and creating an environment where they can thrive, be challenged, and perform their best work.

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

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