• Job Description

  Req#: 5806088003

  Cerebras has developed a radically new chip and system to dramatically accelerate deep learning applications. Our system runs training and inference workloads orders of magnitude faster than contemporary machines, fundamentally changing the way ML researchers work and pursue AI innovation.

  We are innovating at every level of the stack – from chip, to microcode, to power delivery and cooling, to new algorithms and network architectures at the cutting edge of ML research. Our fully-integrated system delivers unprecedented performance because it is built from the ground up for deep learning workloads.

  The role

  As an applied machine learning engineer, you will work on adapting state of the art deep learning (DL) models to run on our wafer scale system. This includes both functional validation and performance tuning of a variety of core models for applications like Natural Language Processing (NLP), Large Language Models (LLMs), Computer Vision (CV) and Graph Neural Networks (GNN).

  • Dataloader implementation and performance optimization
  • Reference model implementation and functional validation
  • Model convergence and hyper-parameters tuning
  • Model customization to meet customer needs.
  • Model architecture pathfinding.

  This role will allow you to work closely with partner companies at the forefront of their fields across many industries. You will get to see how deep learning is being applied to some of the world’s most difficult problems today and help ML researchers in these fields to innovate more rapidly and in ways that are not currently possible on other hardware systems.

  Responsibilities

  • Analyze, implement, and optimize DL models for the WSE
  • Functional and convergence of models on the WSE
  • Work with engineering teams to optimize models for the Cerebras stack
  • Support engineering teams in functional and performance scoping new models and layers
  • Work with customers to optimize their models for the Cerebras stack.
  • Develop new approaches for solving real world AI problems on various domains.

  Requirements

  • Bachelor's degree in engineering, science, or related field with 8+ years of experience
  • Experience programming in modern language like Python or C++
  • In-depth understanding of DL learning methods and model architectures
  • Experience with DL frameworks like PyTorch, TensorFlow and JAX
  • Familiar with state-of-the-art transformer architectures for language and vision model.
  • Experience in model training and hyper-parameter tuning techniques.
  • Familiar with different LLM downstream tasks and datasets.

  Preferred Skills

  • A deep passion for cutting edge artificial intelligence techniques
  • Master's or PhD in engineering, science or related field
  • Understanding of hardware architecture
  • Experience programming accelerators like GPUs and FPGAs

  Why Join Cerebras

  People who are serious about software make their own hardware. At Cerebras we have built a breakthrough architecture that is unlocking new opportunities for the AI industry. With dozens of model releases and rapid growth, we’ve reached an inflection point in our business. Members of our team tell us there are five main reasons they joined Cerebras:

  1. Build a breakthrough AI platform beyond the constraints of the GPU
  2. Publish and open source their cutting-edge AI research
  3. Work on one of the fastest AI supercomputers in the world
  4. Enjoy job stability with startup vitality
  5. Our simple, non-corporate work culture that respects individual beliefs

  Read our blog: Five Reasons to Join Cerebras in 2024.

  Apply today and become part of the forefront of groundbreaking advancements in AI.

  Cerebras Systems is committed to creating an equal and diverse environment and is proud to be an equal opportunity employer. We celebrate different backgrounds, perspectives, and skills. We believe inclusive teams build better products and companies. We try every day to build a work environment that empowers people to do their best work through continuous learning, growth and support of those around them.

  This website or its third-party tools process personal data. For more details, click here to review our CCPA disclosure notice.

• About the company

  Cerebras Systems is an American artificial intelligence company with offices in Sunnyvale and San Diego, California, Toronto, and Tokyo.

Notice