At a Glance
- Tasks: Adapt and optimize language and vision models for cutting-edge AI hardware.
- Company: Join Cerebras, a leader in AI technology with groundbreaking innovations.
- Benefits: Hybrid work model, competitive salary, and opportunities for professional growth.
- Other info: Collaborative environment with exposure to top-tier AI projects and research.
- Why this job: Be at the forefront of AI research and make a real impact on technology.
- Qualifications: Experience in machine learning and strong programming skills in Python or C++.
The predicted salary is between 167400 - 204600 $ per year.
This role adapts and optimizes language and vision models for inference on Cerebras hardware. You will work with researchers and engineers on inference methods such as speculative decoding, model compression, and sparse attention, taking projects from research and prototyping through validation and optimization.
- Permanent
- Hybrid
- 2–7+ years, depending on education
- Bachelor's, Master's, or PhD in a related technical field
Skills and expertise
- Python
- C++
- PyTorch
- Transformers
- vLLM
- SGLang
- Triton
- CUDA
Key responsibilities
- Implement and optimize transformer architectures for language and vision inference on Cerebras hardware.
- Prototype inference algorithms involving speculative decoding, pruning, compression, and sparse attention.
- Train, validate, profile, and optimize models and tools for inference performance.
Minimum Qualifications
- One of the following education and experience combinations:
- Bachelor’s degree in Computer Science, Software Engineering, Computer Engineering, Electrical Engineering, or a related technical field AND 7+ years of ML software development experience
- Master’s degree in Computer Science or related technical field AND 4+ years of software development experience
- PhD in Computer Science or related technical field with 2+ years of relevant research or industry experience
- Equivalent practical experience.
- 4+ years of experience testing, maintaining, or launching software products, including 2+ years of experience with software design and architecture.
- 3+ years of experience in software development focused on machine learning (e.g., deep learning, large language models, or computer vision).
- Strong programming skills in Python and/or C++.
- Experience with Generative AI and Machine Learning systems.
- Evidence of research impact in machine learning, such as publications at top conferences (NeurIPS, ICLR, ICML, ACL, EMNLP, MLSys) or comparable contributions to widely used open-source projects or high-quality preprints.
Preferred Qualifications
- Master’s degree or PhD in Computer Science, Computer Engineering, or a related technical field.
- Experience independently driving complex ML or inference projects from prototype to production-quality implementations.
- Hands-on experience with relevant ML frameworks such as PyTorch, Transformers, vLLM, or SGLang.
- Experience with large language models, mixture-of-experts models, multimodal learning, or AI agents.
Senior Research Engineer employer: We Love AI Jobs
Cerebras Systems is an exceptional employer for Senior Research Engineers, offering a dynamic hybrid work environment that fosters innovation and collaboration. With a focus on cutting-edge AI technology, employees benefit from opportunities for professional growth, access to state-of-the-art resources, and the chance to work alongside industry leaders in machine learning. The company's commitment to pushing the boundaries of AI ensures that team members are engaged in meaningful projects that have a real impact on the future of technology.
