ML Engineer/Researcher
San Francisco, CA at a glance
- Rent
- #2 of 51$2,680/mo+46% vs US avg
- Weather
- #17 of 51295 mild days0 hot · 0 cold
- Income tax
- #1 of 5113.3% top rateCalifornia
What you need
- Extensive PyTorch experience training diffusion, GANs, language models
- JavaScript (React/NextJS) and backend development experience
- Audio model experience a huge plus
- Audiophile or music production experience a huge plus
- Open-source generative model contributions count
What you'll do
- Lead research to improve generative music model quality
- Productize research results into production
- Implement and improve generative architectures (diffusion, GANs, LLMs)
- Run distributed training on hundreds of H100 GPUs
- Collaborate on PyTorch-based model development
Languages: Python (PyTorch), JavaScript (TypeScript)
Listen and try Sonauto: https://sonauto.ai/
About the role:
We're training state of the art generative models in a completely new modality where the standard of quality is significantly higher than those that came before (hands with three fingers is fine in images, but miss a beat in a song and it's ruined). Accordingly, we are consistently implementing new generative architectures and improving current ones.
Right now we're doing distributed training at the scale of hundreds of H100s on diffusion models, GANs, language models, and more written in PyTorch.
You will lead research efforts for improving the song and audio quality of our generative music models, along with productizing the results.
We’re looking for people with:
- Extensive experience with PyTorch training generative models like diffusion models, GANs, and language models. Work with open-source orgs/projects counts too.
- Experience with JavaScript (our frontend is React/NextJS) and backend development are also important to us.
- Experience with audio models is a huge plus.
- Many of our biggest improvements wouldn’t have been possible without extremely close listening, so being an audiophile and/or having music production experience are also huge pluses.
Optimize your resume for this job
Get a match score and the keywords you're missing
About Treblo
Treblo is an AI music editor that turns prompts, lyrics, or melodies into full songs in any style. People use Treblo to have fun and express themselves, and some have even made Hot 100 hits.
Similar Research roles
