Staff Engineer - Training & Inference for Distributed AI

Boson AI; Santa Clara (CA); On-site; USD 150,000 - 600,000; Full time

Job summary

A pioneering AI startup located in Santa Clara is looking for research scientists and engineers to join their team. The role involves optimizing model architectures and implementing distributed optimization algorithms. The ideal candidates should have a strong background in CUDA, PyTorch, and deep learning architectures, with at least a Master’s or Doctoral degree in computer science. Competitive salary ranges from $150,000 to $600,000 per year.

Qualifications

• Experience in writing clean and efficient code.
• Proficiency in at least one deep learning framework.
• Participation in at least 1 research project related to distributed training.

Responsibilities

• Optimize model architectures to handle images, video, text, speech, and audio data.
• Implement and optimize kernels for efficient training on GPUs.
• Conduct performance optimization and distributed training.

Skills

CUDA

PyTorch

Distributed optimization

Deep learning architectures

Performance optimization

Education

Master or Doctoral degree in computer science or equivalent

Tools

Triton

JAX

Job description

A pioneering AI startup located in Santa Clara is looking for research scientists and engineers to join their team. The role involves optimizing model architectures and implementing distributed optimization algorithms. The ideal candidates should have a strong background in CUDA, PyTorch, and deep learning architectures, with at least a Master’s or Doctoral degree in computer science. Competitive salary ranges from $150,000 to $600,000 per year.