Senior AI Research Scientist - LLMs & Scalable Training

Sciforium; San Francisco (CA); On-site; USD 130,000 - 160,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401k plan

Daily lunch, snacks, and beverages

Flexible time off

Competitive salary and equity

Job summary

An AI infrastructure company is seeking a Senior Research Scientist specializing in advanced AI/ML to lead cutting-edge research in large language models and generative media. The ideal candidate will have a PhD and over 5 years of relevant experience, with strong coding skills in Python and expertise in frameworks like PyTorch and JAX. This role offers competitive salary, equity, and comprehensive benefits including healthcare and flexible time off.

Qualifications

• PhD is required.
• 5+ years of research experience or exceptional fresh PhDs accepted.
• Proven track record of impactful research.

Responsibilities

• Lead research in advanced machine learning areas including LLMs and generative AI.
• Design and evaluate ML algorithms using frameworks like PyTorch and JAX.
• Conduct large-scale distributed training experiments.

Skills

Python

PyTorch

JAX

Machine Learning

Deep Learning

Distributed Training

Education

PhD in Computer Science, Machine Learning, AI, Mathematics, or a related field

Tools

Flax

XLA

Job description

An AI infrastructure company is seeking a Senior Research Scientist specializing in advanced AI/ML to lead cutting-edge research in large language models and generative media. The ideal candidate will have a PhD and over 5 years of relevant experience, with strong coding skills in Python and expertise in frameworks like PyTorch and JAX. This role offers competitive salary, equity, and comprehensive benefits including healthcare and flexible time off.