Cerebras is developing a radically new chip and system to dramatically accelerate deep learning applications. Our system runs training and inference workloads orders of magnitude faster than contemporary machines, fundamentally changing the way ML researchers work and pursue AI innovation.
We are innovating at every level of the stack – from chip, to microcode, to power delivery and cooling, to new algorithms and network architectures at the cutting edge of ML research. Our fully-integrated system delivers unprecedented performance because it is built from the ground up for the deep learning workload.
Cerebras is building a team of exceptional people to work together on big problems. Join us!
About The Role
As a Frameworks Software Engineer on our team, you will work with leaders from industry and academia at the intersection of hardware and software, to develop state-of-the-art solutions for emerging problems in AI compute.
The Cerebras software platform is designed to be targeted by today’s most relevant machine learning frameworks, such as TensorFlow, PyTorch, Caffe2, and MXNet. Our ML software engineers are responsible for integrating these frameworks to work with our own highly optimized software stack. Fundamentally, you will be enabling ML researchers to use the software tools and workflows of today to unlock the advanced hardware capabilities of tomorrow.
Responsibilities
- Create tools and design workflows that enable the development, training, and deployment of machine learning models on our new hardware system
- Map abstract computations expressed via third-party ML frameworks into representations that can then be compiled into highly optimized executables that target Cerebras’ system
- Develop connections between representations of existing deep learning frameworks -- such as TensorFlow, Caffe/2, MXNet, CNTK -- with our customized back-end
- Understand the runtime environments of existing frameworks and our backend, and develop an execution model connecting them together in a way that is seamless to the user
Requirements
- Bachelors, Masters, PhD degree or foreign equivalent in Computer Science, Engineering, or related field
- Understanding of state-of-the-art deep learning model architectures and training protocols
- Direct experience with one ML framework internals (like TensorFlow, PyTorch, ONNX, etc) strongly preferred
- Strong Python and C++ development skills
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.
Jobs from our Partners
Application Developer
Senior System Integration and Test Engineer
Data Engineer I – Enterprise Payment and Billing
Sr. Data Engineer - Artificial Intelligence
Junior Java Developer
Other Jobs from CerebrasSystems
IT/DevOps Engineer
AI Compute Infrastructure Engineer
Network Engineer
Senior Applied ML Engineer
Software Automation Engineer
Principal Product Manager, ML
Similar Jobs
Machine Learning Engineer
Senior Data Scientist
Senior Machine Learning Engineer
Senior Staff Machine Learning Engineer, Trust
There are more than 50,000 engineering jobs:
Subscribe to membership and unlock all jobs
Engineering Jobs
50,000+ jobs from 4,500+ well-funded companies
Updated Daily
New jobs are added every day as companies post them
Refined Search
Use filters like skill, location, etc to narrow results
Become a member
🥳🥳🥳 212 happy customers and counting...
Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.
Cancel anytime / Money-back guarantee