Xai
Sr. Network Engineer, High-Speed Interconnects (ML Infrastructure)
Palo Alto, CA. Posted 9 months ago
The role
You will design, validate and productize high-speed copper and optical interconnects for large AI training and inference clusters. You will own the network fabric work from design and build through operations, and work closely with vendors and ML training teams.
What you will do
- Design, validate and productize high-speed copper and optical connectivity for AI clusters at 100k+ GPU scale
- Own vendor due diligence and onboarding for 1.6T products, including AEC and pluggable optical transceivers, with bring-up and characterization
- Investigate LPO and LRO in the network
- Evaluate early co-packaged and near-packaged engines for switches and GPUs
- Evaluate new interconnect modalities such as VCSEL, microLED and THz radio solutions
- Work with vendors on transceiver, cable, SerDes, DSP and silicon photonics roadmaps
- Work with ML training teams to turn communication patterns into interconnect topology requirements
- Perform system-level simulation of end-to-end fabric performance
- Drive failure analysis and corrective actions for interconnect issues in production clusters
- Build internal tooling and automation for interconnect health monitoring, diagnostics and qualification
Required qualifications
- 5+ years of experience designing, deploying and operating high-speed copper and optical interconnects
- Master's degree or Ph.D. in electrical engineering, photonics, or physics
Nice to have
- Experience in a module design role or in a hyperscale datacenter environment
- Deep knowledge of PAM4 SerDes performance, equalization, jitter and crosstalk
- Solid operational understanding of FEC, retimers, TIAs and drivers
- Deep knowledge of optical link budget analysis and metrics such as TDECQ, OMA and stressed receiver sensitivity
- Expertise in transceiver components such as CW lasers, SiPh PICs, EML, DSP and their failure modes
- Knowledge of thermal, mechanical, power and signal integrity constraints in dense hardware
- Knowledge of SiPh design process, yield improvement and reliability testing
- Familiarity with CPO technologies and their risk areas
- Familiarity with subcomponent supply chains and global manufacturers, ODMs and CMs
- Strong problem-solving skills and ability to thrive in a fast-paced, ambiguous setting
Benefits
- Equity
- Comprehensive medical coverage
- Vision coverage
- Dental coverage
- 401(k) retirement plan
- Short-term and long-term disability insurance
- Life insurance
- Various discounts and perks
Before you apply
- Master's degree or Ph.D. in electrical engineering, photonics, or physics is required
About the company
SpaceXSI builds Super Intelligence, including Colossus supercomputers, Grok models and applications such as Grok Bot, Grok Build, Grok Imagine, Grok Voice and X. The company states its goal is to help humanity understand the universe.