Netpreme · 1 month ago
Member of Technical Staff, System Software
Santa Clara, CA
Full-time
Onsite
Mid, Senior Levelnetpreme is seeking experienced low-level systems engineers to design device drivers and systems software for emerging ML hardware. The role involves collaboration with Hardware and ML Software teams to create efficient low-level system layers for communication devices.
Computer Software
Responsibilities
Develop low-latency, high-throughput data exchange systems between GPUs
Develop high-performance data movement kernels
Define and expose data movement interfaces to high-level ML frameworks (e.g. PyTorch)
Develop Linux device drivers for custom hardware
Qualification
Required
Hacker mentality
Deep experience with low-level systems programming
Knowledge of OS internals, especially PCIe/IO sub-systems and memory management
Prior experience in accelerator programming (e.g. CUDA, JAX/Pallas, ROCm)
Prior experience with collective communication libraries (e.g. nccl)
Preferred
Experience with GPUDirect and RDMA is a strong plus
Benefits
Comprehensive benefits including health, dental, vision, and life insurance.
Relocation assistance and visa sponsorship.
Perks include a daily lunch stipend, 401k match, and more.
Company
Netpreme
Empowering AI Systems with Supreme Networking
Founded in 2024
11-50 employees
Funding
Current Stage
Early StageCompany data provided by crunchbase