Cerebras
Software Engineer - Host and Network IO
Sunnyvale, CA, Remote. Posted 2 months ago
The role
This person will develop the full IO path implementation between distributed server nodes and custom hardware. They will interface between AI application, cluster architecture, and FPGA/ASIC teams to optimize bandwidth and latency.
What you will do
- Develop x86 & ARM software to expose next-generation hardware IO capabilities
- Govern a generic IO API with multiple internal users
- Develop control and configuration subsystems interacting with Cerebras hardware
- Drive network performance debug of large AI clusters
- Gather and analyze network statistics and packet traces to root cause bottlenecks
- Develop tools/telemetry for increasing visibility into the network and IO datapath
- Optimize cpu/mem utilization leveraging kernel bypass and zero-copy techniques
- Integrate leading edge networking technologies and protocols
- Lead cross-functional technical projects spanning multiple teams
- Foster clear and effective communication across teams and stakeholders
Required qualifications
- Master's/PhD in Computer Science or Electrical Engineering + 1 year industry experience, OR 3+ years industry experience
- Experience in large software environments
- Embedded systems, HW/SW co-design, and some driver development
- Network protocol familiarity (TCP, RoCE)
Nice to have
- network debug tools such as Wireshark, or willingness to learn
- Some network switch environment familiarity (Arista, Juniper, etc.) or willingness to learn
Benefits
- Build a breakthrough AI platform beyond the constraints of the GPU
- Publish and open source their cutting-edge AI research
- Work on one of the fastest AI supercomputers in the world
- Enjoy job stability with startup vitality
- Simple, non-corporate work culture that respects individual beliefs
About the company
Cerebras Systems builds the world's largest AI chip. The company works with leading model labs, global enterprises, and cutting-edge AI-native startups.