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.