Apply on Employer Site

Boson AI · 18 hours ago

Member of Technical Staff, Training and Inference

Santa Clara, CA

Full-time

Onsite

Mid Level

$150K/yr - $600K/yr

Boson AI is an early-stage startup building large audio models for everyone to enjoy and use. They are seeking research scientists and engineers to optimize model architectures and implement distributed optimization algorithms in their Santa Clara office.

Artificial Intelligence (AI)Information TechnologyMarket Research

H1B Sponsor Likely

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

Qualification

CUDAPyTorchDistributed optimizationDeep learning architecturesTritonClean codeComputer networkingData handling

Required

Strong background in CUDA, Triton, PyTorch, distributed optimization and deep learning architectures

Master or Doctoral degree in computer science or equivalent

Proficiency in at least one deep learning framework, such as PyTorch or JAX

Experience in writing clean and efficient code

Participated in at least 1 research project related to distributed training or inference

Preferred

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

Company

Boson AI

Boson AI is an AI company that develops large language model tools.

Founded in 2023

Santa Clara, California, USA

11-50 employees

https://boson.ai/

H1B Sponsorship

Boson AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (4)

2024 (7)

2023 (2)

Funding

Current Stage

Early Stage

Leadership Team

Alex Smola

CEO & Cofounder

Mu Li

Co-Founder & CTO

Company data provided by crunchbase