DriveNets

GPU Team Leader

Tel Aviv, Israel
C C++ Triton CUDA HIP OpenCL PyTorch TensorFlow JAX Docker Kubernetes
Description

GPU Team Leader

Location: Tel Aviv, Israel (IL)

Description

Location: Tel Aviv

#Hybrid

DriveNets is a leader in high-scale disaggregated networking solutions. Founded in 2015, DriveNets modernizes the way service providers, cloud providers and hyperscalers build networks. Supporting the largest network in the world, more than half of AT&T’s backbone traffic is running on DriveNets’ Network Cloud open disaggregated architecture. Raising $587 million in three funding rounds, DriveNets is disrupting the networking market from high-scale architecture to AI platforms, and is bringing onboard the most talented people. We are seeking people that want to make an impact on the world’s leading communication networks and are experienced in networking architecture or AI infrastructure solutions.

Job Summary

We are seeking an experienced technical leader to head our collective communication library development team. This role involves leading a team of engineers in developing high-performance collective communication implementations for multi-NPU and multi-node AI workloads.

Key Responsibilities

  • Lead the design and development of collective communication primitives (All-Reduce, All-to-All, Gather/Scatter and etc)
  • Architect scalable communication protocols for multi-NPU and multi-node systems
  • Optimize communication performance for NPU architectures
  • Provide technical leadership to the team members in NPU programming, distributed systems, and communication protocols
  • Work with a success-driven worldwide international team (Network, NPU, QA, AI, DL/ML Framework)
  • Define project milestones, deliverables, and technical roadmaps
  • Ensure compatibility with major AI frameworks (PyTorch, TensorFlow, JAX) 

Requirements

Required Qualifications

  • BSc/MSc in computer science/computer engineering or equivalent
  • 8+ years of experience in systems programming and distributed computing
  • 5+ years of leadership experience managing technical teams
  • Expert-level C/C++ programming with focus on performance optimization
  • Experience with NPU programming (Triton / CUDA / HIP / OpenCL)
  • Deep understanding of distributed systems, communication protocols, and network programming
  • Experience with DL/ML frameworks (PyTorch, TensorFlow) and distributed training / inferencing
  • Experience with performance profiling and optimization tools
  • Strong communication and interpersonal skills

Preferred Qualifications 

  • Experience with NPU communication library development
  • Contributions to open-source projects (PyTorch, TensorFlow, communication libraries)
  • Familiarity with containerization and orchestration
  • Interoperability experience with partners, vendors and external teams


DriveNets
DriveNets

0 applies

0 views

There are more than 50,000 engineering jobs:

Subscribe to membership and unlock all jobs

Engineering Jobs

60,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

🥳🥳🥳 452 happy customers and counting...

Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.

To try it out

For active job seekers

For those who are passive looking

Cancel anytime

Frequently Asked Questions

  • We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
  • We've got over 200,000 jobs from 15,000+ vetted companies. No fake or sleazy jobs here!
  • We aggregate jobs from 15,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
  • We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
  • Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
  • Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
  • Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅

What Fellow Engineers Say