Boson AI
Member of Technical Staff, Training and Inference
Boson AI, Santa Clara, California, us, 95053
Overview
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 tuned 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. Base pay range $150,000.00/yr - $600,000.00/yr 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 Qualifications
Experience in writing clean and efficient code; Master or Doctoral degree in computer science or equivalent Proficiency in at least one deep learning framework, such as PyTorch or JAX Participation 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 Other
Referrals increase your chances of interviewing at Boson AI by 2x. Get notified about new Member of Technical Staff jobs in
Santa Clara, CA .
#J-18808-Ljbffr
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 tuned 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. Base pay range $150,000.00/yr - $600,000.00/yr 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 Qualifications
Experience in writing clean and efficient code; Master or Doctoral degree in computer science or equivalent Proficiency in at least one deep learning framework, such as PyTorch or JAX Participation 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 Other
Referrals increase your chances of interviewing at Boson AI by 2x. Get notified about new Member of Technical Staff jobs in
Santa Clara, CA .
#J-18808-Ljbffr