echojobs.io
Login

Boson AI

Member of Technical Staff, Training and Inference

Santa Clara, CA
CUDA Triton PyTorch Deep Learning JAX Infiniband SHARP
Search for More Jobs
Description

Member of Technical Staff, Training and Inference

Team: Engineering

Location: Santa Clara HQ

Commitment: Full-time

Workplace Type: onsite

Salary:

Total compensations includes base pay, equity, and benefits.

Boson AI is an early-stage startup building large audio models for everyone to enjoy and use. Our founders (Alex Smola,Mu Li), and a team of Deep Learning, Optimization, NLP, and Statistics scientists and engineers are working on high quality generative AI models for language and beyond.

We are seeking research scientists and engineers to join our team full-time in our Santa Clara office. As part of your role, you will work on implementing and improving distributed optimization algorithms, performance tune architectures, improve inference and help us make Deep Networks perform efficiently on our cluster. The ideal candidate will possess a strong background in CUDA, Triton, PyTorch, distributed optimization and deep learning architectures.

We encourage you to apply even if you do not believe you meet every single qualification. As long as you are motivated to learn and join the development of foundation models, we’d love to chat.

Responsibilities

  • Optimize model architectures and loss objectives to handle combinations of images, video, text, speech, and audio data.
  • Implement and optimize kernels for efficient training on Hopper and Blackwell GPUs
  • Performance optimization (floating point formats, sparsity, systems level optimization)
  • Distributed optimization and training

You may be a good fit if you have:

  • Experience in writing clean and efficient codeMaster or Doctoral degree in computer science or equivalent.
  • Proficiency in at least one deep learning framework, such as PyTorch or JAX.
  • Participated in at least 1 research project related to distributed training or inference.

Strong candidates may also have:

  • Experience in implementing your own kernels in CUDA or another compiler/toolkit (Triton, ThunderKittens, PTX, etc.)
  • Experience in distributed optimization (e.g. using DeepSpeed, FSDP), ideally designing performance optimizations
  • Experience in computer networking (e.g. Infiniband, using SHARP)
  • Experience in handling data at billions-scale
Boson AI
Boson AI

0 applies

0 views

There are more than 50,000 engineering jobs:

Subscribe to membership and unlock all jobs

Engineering Jobs

60,000+ jobs from 4,500+ well-funded companies

Updated Daily

New jobs are added every day as companies post them

Refined Search

Use filters like skill, location, etc to narrow results

Become a member

🥳🥳🥳 452 happy customers and counting...

Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.

To try it out

For active job seekers

For those who are passive looking

Cancel anytime

Frequently Asked Questions

  • We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
  • We've got over 200,000 jobs from 15,000+ vetted companies. No fake or sleazy jobs here!
  • We aggregate jobs from 15,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
  • We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
  • Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
  • Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
  • Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅

What Fellow Engineers Say