ML Research Engineer (Inference)

Reposted 10 Days Ago
Bengaluru, Bengaluru Urban, Karnataka, IND
In-Office
Junior
Artificial Intelligence
The Role
As an ML Research Engineer, you will adapt and optimize language and vision models for efficient inference on Cerebras hardware, focusing on advanced techniques for low-latency, high-throughput performance.
Summary Generated by Built In

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. Our novel wafer-scale architecture provides the AI compute power of dozens of GPUs on a single chip, with the programming simplicity of a single device. This approach allows Cerebras to deliver industry-leading training and inference speeds and empowers machine learning users to effortlessly run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs.  

Cerebras' current customers include top model labs, global enterprises, and cutting-edge AI-native startups. OpenAI recently announced a multi-year partnership with Cerebras, to deploy 750 megawatts of scale, transforming key workloads with ultra high-speed inference. 

Thanks to the groundbreaking wafer-scale architecture, Cerebras Inference offers the fastest Generative AI inference solution in the world, over 10 times faster than GPU-based hyperscale cloud inference services. This order of magnitude increase in speed is transforming the user experience of AI applications, unlocking real-time iteration and increasing intelligence via additional agentic computation.

About The Role 

As a Research Engineer on the Inference ML team at Cerebras Systems, you will adapt today's most advanced language and vision models to run efficiently on our flagship Cerebras architecture. You'll work alongside ML researchers and engineers to design, prototype, validate, and optimize models, gaining end-to-end exposure to cutting-edge inference research on the world's fastest AI accelerator. 

You will focus on pushing the frontier of speculative decodinglarge-model pruning and compressionsparse attention, and sparsity-driven techniques to deliver low-latency, high-throughput inference at scale. 

Responsibilities 

  • Implement and adapt transformer-based models (NLP and/or vision) to run on Cerebras hardware
  • Assist in optimizing models for inference performance (latency, throughput)
  • Run experiments, analyze results, and support model improvements
  • Help bring up and validate models on the Cerebras system
  • Debug and troubleshoot model or system issues with guidance from senior team members
  • Support profiling and performance analysis using internal tools
  • Collaborate with cross-functional teams (ML, software, hardware) on model integration

Minimum Qualifications 

  • Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
  • 1–3 years of experience in software engineering or machine learning in a similar capacity (internships count)
  • Experience with Python and at least one ML framework (e.g., PyTorch, Transformers, vLLM or SGLang)
  • Understanding of deep learning concepts (e.g., neural networks, transformers)
  • Experience with Generative AI and Machine Learning systems
  • Strong programming skills in Python and/or C++

Preferred Qualifications 

  • Experience with speculative decoding, neural network pruning and compression, sparse attention, quantization, sparsity, post-training techniques, and inference-focused evaluations. 
  • Exposure to large language models or computer vision models
  • Experience running experiments or tuning models
  • Familiarity with tools like PyTorch, Hugging Face Transformers, or similar
  • Basic understanding of performance concepts (e.g., latency, throughput)
  • Experience working in Linux environments
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 2026.

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.

Skills Required

  • Bachelor's or Master's degree in Computer Science, Engineering, or related field
  • 1-3 years of experience in software engineering or machine learning
  • Experience with Python and at least one ML framework
  • Understanding of deep learning concepts
  • Experience with Generative AI and Machine Learning systems
  • Strong programming skills in Python and/or C++

Cerebras Systems Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Cerebras Systems and has not been reviewed or approved by Cerebras Systems.

  • Fair & Transparent Compensation Pay is considered competitive for an AI‑hardware firm, and many employees are described as generally happy with compensation. Sentiment indicates compensation is viewed favorably while acknowledging variation by role and seniority.
  • Healthcare Strength Health coverage is described as top quality with medical, dental, and vision included. Premiums are reportedly fully covered for employees in some plans, increasing perceived value.
  • Flexible Benefits Work‑from‑home flexibility is regarded as strong. Flexible arrangements complement standard offerings like vacation, sick leave, and paid holidays.

Cerebras Systems Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Sunnyvale, CA
402 Employees
Year Founded: 2016

What We Do

Cerebras Systems is a team of pioneering computer architects, computer scientists, deep learning researchers, functional business experts and engineers of all types. We have come together to build a new class of computer to accelerate artificial intelligence work by three orders of magnitude beyond the current state of the art. The CS-2 is the fastest AI computer in existence. It contains a collection of industry firsts, including the Cerebras Wafer Scale Engine (WSE-2). The WSE-2 is the largest chip ever built. It contains 2.6 trillion transistors and covers more than 46,225 square millimeters of silicon. The largest graphics processor on the market has 54 billion transistors and covers 815 square millimeters. In artificial intelligence work, large chips process information more quickly producing answers in less time. As a result, neural networks that in the past took months to train, can now train in minutes on the Cerebras CS-2 powered by the WSE-2. Join us: https://cerebras.net/careers/

Similar Jobs

Micron Technology Logo Micron Technology

Staff Engineer

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
45000 Employees
5-5 Annually

Micron Technology Logo Micron Technology

Staff Engineer

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
45000 Employees

Take-Two Interactive Software Logo Take-Two Interactive Software

Talent Acquisition Coordinator

Gaming • Information Technology • Mobile • Software
Hybrid
Bengaluru, Bengaluru Urban, Karnataka, IND
13000 Employees

GitLab Logo GitLab

Senior Program Manager

Cloud • Security • Software • Cybersecurity • Automation
Easy Apply
In-Office or Remote
Bangalore, Bengaluru Urban, Karnataka, IND
2500 Employees

Similar Companies Hiring

Legora Thumbnail
Artificial Intelligence • Legal Tech • Software
Chicago, Illinois
700 Employees
Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account