Machine Learning Engineer
New York or London
Andreessen Horowitz
Andreessen Horowitz (a16z) is a venture capital firm in Silicon Valley, California, investing in bold founders, innovators, and entrepreneurs building the future through technology.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 CompetenciesThe 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++
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
- 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.
Tags: ChatGPT Copilot CUDA Deep Learning Engineering Gemini Generative AI Generative modeling GPT GPU JAX LLMs Machine Learning Model training Open Source PyTorch TensorFlow
Perks/benefits: Career development Competitive pay Equity / stock options Flex vacation Health care Insurance
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Research Scientist jobs
- Open Data Science Manager jobs
- Open Data Engineer II jobs
- Open Principal Data Scientist jobs
- Open Business Data Analyst jobs
- Open BI Analyst jobs
- Open Sr Data Engineer jobs
- Open Data Scientist II jobs
- Open Business Intelligence Engineer jobs
- Open Data Science Intern jobs
- Open Sr. Data Scientist jobs
- Open Senior Business Intelligence Analyst jobs
- Open Software Engineer, Machine Learning jobs
- Open Lead Data Analyst jobs
- Open Azure Data Engineer jobs
- Open Junior Data Scientist jobs
- Open MLOps Engineer jobs
- Open Data Analytics Engineer jobs
- Open Manager, Data Engineering jobs
- Open Marketing Data Analyst jobs
- Open Data Engineer III jobs
- Open Junior Data Engineer jobs
- Open Data Engineering Manager jobs
- Open Data Analyst II jobs
- Open ETL Developer jobs
- Open Data quality-related jobs
- Open Data management-related jobs
- Open Privacy-related jobs
- Open Excel-related jobs
- Open ML models-related jobs
- Open Data pipelines-related jobs
- Open APIs-related jobs
- Open PhD-related jobs
- Open PyTorch-related jobs
- Open LLMs-related jobs
- Open Finance-related jobs
- Open Data visualization-related jobs
- Open Consulting-related jobs
- Open TensorFlow-related jobs
- Open Deep Learning-related jobs
- Open Business Intelligence-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open NLP-related jobs
- Open Data governance-related jobs
- Open DevOps-related jobs
- Open Kubernetes-related jobs
- Open Git-related jobs
- Open Hadoop-related jobs
- Open Docker-related jobs