C

Cerebras

AI accelerator hardware replacing GPUs

Applied Machine Learning Research Scientist

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeNo salary listedHybrid

About the Job

Requirements

  • A bachelor's or master's degree in Computer Science, Engineering, or a related field.
  • At least 4 years of experience working with machine-learning systems, including internship, research, or industry experience.
  • Strong programming skills in Python.
  • Experience with machine-learning frameworks such as PyTorch.
  • A solid understanding of machine-learning fundamentals.
  • Familiarity with deep-learning architectures, particularly transformers.
  • Ability to read and understand modern machine-learning papers and implement key ideas.

Responsibilities

  • Apply post-training techniques such as RLVR, RLHF, and GRPO to improve model performance.
  • Build and maintain evaluation pipelines to measure model performance across tasks and domains.
  • Debug issues across the machine-learning stack, including data pipelines, training jobs, model outputs, and mixed- or lower-precision computation.
  • Collaborate with researchers to translate machine-learning ideas into efficient, scalable implementations.
  • Design, implement, and scale machine-learning pipelines across all stages of large language model development, including pretraining, fine-tuning, and alignment.
  • Work with large datasets, including dataset generation, filtering, and synthetic-data approaches.
  • Optimize training and inference workflows for performance, efficiency, and reliability.
  • Contribute high-quality, maintainable code to shared machine-learning infrastructure.

Desired Qualifications

  • Experience working with large language models, including training, fine-tuning, and evaluation.
  • Familiarity with reinforcement-learning concepts.
  • Experience with distributed-training frameworks such as FSDP and Megatron.
  • Experience working with large-scale datasets and data pipelines.
  • Experience debugging or optimizing machine-learning systems for performance.
  • Contributions to meaningful codebases, projects, or open-source systems.

About the Company

Cerebras

AI accelerator hardware replacing GPUs

Cerebras Systems creates AI acceleration hardware and software. Its CS-2 system is designed to replace traditional GPU clusters for AI workloads, speeding up training and inference while simplifying the setup by eliminating the need for parallel programming, distributed training, and cluster management. The product works as a single, large processor-based accelerator with accompanying software and cloud services to run AI models efficiently, reducing latency and time to results. Compared with competitors, Cerebras differentiates itself with the largest processor in the industry and an integrated hardware-software stack that aims to streamline AI workflows rather than relying on multi-GPU clusters. The company's goal is to help research labs, healthcare, finance, and other industries achieve faster, more cost-effective AI development and deployment by offering a turnkey high-performance AI compute solution.

Company size
501-1,000
Company stage
IPO
Headquarters
Sunnyvale, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at Cerebras

See more jobs