Machine Learning Engineer - Music Platform - SF Job at Benchstack Ai in San Fra

Machine Learning Engineer - Music Platform - SF

This range is provided by Benchstack Ai. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$150,000.00/yr - $300,000.00/yr

We’re hiring a Machine Learning Engineer / Researcher with experience training large-scale generative models (diffusion, GANs, or language models) using PyTorch and distributed infrastructure. You’ll help build next-generation music generation models where precision and audio quality matter more than anything — one off-beat and the product fails.

You’ll be leading hands-on research and large-scale training on clusters of hundreds of H100s, improving the song and audio quality of our generative systems and bringing those breakthroughs into production.

What you’ll do

• Train and optimise large-scale generative models across multiple modalities
• Research and deploy new architectures that push the boundaries of audio generation
• Collaborate directly with the founders on product integration and scaling
• Own distributed training pipelines end-to-end

What we’re looking for

• 1–4 years post graduation experience training generative models (diffusion, GANs, transformers, etc.)
• Deep experience with PyTorch and distributed compute
• Solid foundations in Python, with some familiarity in TypeScript
• Driven by research curiosity and a builder mindset

Nice to have

• Background in audio or music generation
• Open-source ML research contributions
• Experience training models at scale (multi-node / multi-GPU setups)

🧠 Visa: TN or H1B transfer only

🌐 Company: 5-person, VC-backed startup ($2.5M raised)

We’re reimagining how music is created and experienced — join a team at the forefront of generative audio.

Seniority level

Associate

Employment type

Full-time

Job function

Engineering, Information Technology, and Product Management