Applied ML Engineer, Cerebras

Applied ML Engineer, Cerebras

Company Cerebras
Job title Applied ML Engineer
Job location UAE
Type Full Time

Responsibilities:

  • Familiar with state-of-the-art transformer architectures for language and vision models.
  • Bring up a new state-of-the-art model on Cerebras System and function validation.
  • Train a model to convergence, and hyper-parameter tuning.
  • Optimize model code to run efficiently on Cerebras System.
  • Explore new model architecture that takes advantage of Cerebras unique capabilities.
  • Develop new approaches for solving real-world AI problems in various domains.

Requirements & Skills:

  • Masters or PhD in Computer Science or related field
  • Familiarity with JAX/TensorFlow/PyTorch
  • Good understanding of how to define custom layers and back-propagate through them.
  • Experience with transformer deep learning models
  • Experience in vertical such as computer vision or language modeling
  • Experience with Large Language Models such as GPT family, Llama, and BLooM.

apply for job button