ML Systems Performance Engineer
Cerebras Systemsabout 4 hours ago
Bengaluru, IndiaMid Level / Senior
H1B Sponsor
Responsibilities
- Build performance models to estimate the performance of ML models.
- Optimize and debug kernel micro code and compiler algorithms.
- Debug and understand runtime performance on the system and cluster.
- Develop tools and infrastructure for visualizing performance data.
Requirements
- Bachelors / Masters / PhD in Electrical Engineering or Computer Science.
- Strong background in computer architecture.
- Exposure to low-level deep learning / LLM math.
- Strong analytical and problem-solving mindset.
- 3+ years of experience in a relevant domain.
- Experience working on CPU/GPU simulators.
- Exposure to performance profiling and debugging.
- Comfort with C++ and Python.
Benefits
- Opportunity to build a breakthrough AI platform.
- Ability to publish and open source cutting-edge AI research.
- Work on one of the fastest AI supercomputers in the world.
- Enjoy job stability with startup vitality.
- Experience a simple, non-corporate work culture.