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.