Member of Technical Staff, Modeling

Boson AI; Toronto; On-site; CAD 80,000 - 150,000; Full time

Job summary

An innovative startup is on the lookout for talented research scientists and engineers to join their dynamic team. This role offers the chance to work on cutting-edge large language models, contributing to the development of state-of-the-art generative AI technologies. You will engage in designing novel model architectures, conducting large-scale experiments, and aligning models to human values. The company fosters a collaborative environment where your scientific contributions can make a significant impact. If you're passionate about advancing AI and eager to tackle complex challenges, this opportunity is perfect for you.

Benefits

401k Plan

HSA

FSA

Free Food

Equity

Qualifications

• Strong background in machine learning with a focus on generative AI.
• Experience in alignment research and large-scale distributed training.

Responsibilities

• Design and verify novel model architectures and training objectives.
• Conduct large-scale experiments to verify modeling choices.

Skills

Machine Learning

Deep Learning

Model Alignment

Problem-Solving

Education

PhD

Master's Degree

Tools

PyTorch

CUDA

Job description

Boson AI is an early-stage startup building large language tools for interaction and entertainment. Our founders, Alex Smola, Mu Li, and a team of Deep Learning, Optimization, NLP, AutoML, 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 modeling and training LLMs, understanding and interpreting model behavior, and aligning models to human values. The ideal candidate will possess a strong background in machine learning and have motivations for developing state-of-the-art models towards AGI.

Responsibilities

1. Design and verify novel model architectures and training objectives.
2. Investigate novel model alignment algorithms.
3. Write efficient and clean code for ML training.
4. Conduct large-scale experiments to verify modeling choices and identify areas for improvement.

Experience

1. Summarize results and clearly communicate motivations and observations in your work.
2. Proficiency in at least one deep learning framework, such as PyTorch.
3. Participation in at least one research project related to LLM or multimodal models, e.g., experience in training or fine-tuning them.
4. Experience in alignment research.
5. Experience in large-scale distributed model training.
6. Experience in writing GPU kernels in CUDA.

Qualifications

1. PhD or Master's degree with solid scientific contributions.
2. Active GitHub repository.
3. Active scientific track record.
4. Excellent problem-solving skills.

Total compensation includes base pay, equity, and benefits. We offer a 401k plan, HSA, FSA, and free food (including dried mangoes).