ML Network Stack Lead Engineer
Amazon Locker
Annapurna Labs, an integral part of AWS, Cupertino, seeks a Lead Software Engineer to own the ML networking stack for EC2 distributed AI/ML systems. You will guide design across ML communication libraries (NCCL, NVSHMEM, NIXL) and CUDA kernels, mentoring engineers and shaping scalable software. You will work on large-scale clusters, collaborate with hardware teams, and drive features for the largest AI models. Requires 5+ years leading design and C/C++ expertise. #J-18808-Ljbffr
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the ML Network Stack Lead Engineer in Cupertino, CA vacancy
$224k - $356.5k
...progress.We’re looking for a Senior Full-Stack Software Engineer to join the AI Hub team within the DGX... ...Superclusters.What You’ll Be Doing:Lead the architecture and delivery of high-scale... ...pairing, and coaching.Stay ahead of AI/ML infrastructure trends and drive...SuggestedFull timeNvidia
Santa Clara, CA4 days ago$185k - $250k
...Knightscope is seeking a Senior Full Stack Engineer to design, build, and own the web platform... ...equivalent cloud services. Integrate network camera APIs and device controls (Axis VAPIX... ...cross-functionally with robotics, AI/ML, audio/video, DevOps, and UI/UX teams; contribute...NetworkFull timeRelocationKnightscope, Inc
Sunnyvale, CA4 hours ago$184k - $287.5k
...seeking a Senior Systems Software test (lead) Engineer to join our Cloud Service Provider (CSP) Engagements team, focusing on ML software stack validation for Datacenter products... ...platforms, firmware, drivers, OS integration, networking, and large-scale cluster environments....NetworkFull timeLocal areaNvidia
Santa Clara, CA3 days ago$225k - $275k
...highly accomplished Principal Engineer - Security Architecture to define... ...ResponsibilitiesDefine and lead the long-term security architecture... ....Familiarity with AI/ML infrastructure security, KV cache... ...infrastructure platforms, including cloud, network, and application security...NetworkDataDirect Networks
Santa Clara, CA1 day ago$193.3k - $261.5k
We are seeking an experienced engineer and technical leader to join our team that owns the network stack for EC2 distributed AI/ML systems. The team develops support for a variety of... ...will be highly regarded. You'll be leading senior, mid-level, and junior SDEs and directing...NetworkLocal areaFlexible hoursAmazon
Cupertino, CA2 days ago$171k - $247k
...timelines, and sustainability goals.Lead the end-to-end physical... ...for ultra-high-density AI/ML platforms (TPUs/GPUs), including... ...outside Architectural and Engineering consultants and synthesize cross... ...requirements (Platforms, Networking, SRE, Construction) into buildable...NetworkWork at officeGoogle
Sunnyvale, CA3 days ago- ...Lead the development of the network stack for EC2 distributed AI/ML systems, specifically enabling NVIDIA GPUs to work with AWS machines. Direct a team of software engineers to deliver features required for the largest AI models and ML workloads. Requirements Requires...NetworkFlexible hours
Amazon
Cupertino, CA5 days ago $150k - $225k
...life. We are looking for a Lead Embedded Software Engineer to own the full path from... ...across the embedded stack. Make and own key technical... ...Implement and optimize networking stacks - BLE, Thread, and Wi... ...Familiarity with TinyML or edge ML frameworks (TFLite Micro, Edge...NetworkFull timeWork at officeImmediate startVisa sponsorshipNight shifteSpace
Saratoga, CA5 days ago- ...Software Engineer Uber Technical Lead, Infrastructure, FMOD Share Software Engineer... ..., distributed systems or networks, or experience with compute... ...problems across the full-stack as we continue to push technology... ...all ACI capacity including ML and Standard Compute...NetworkTemporary workWorldwide
Google LLC
Sunnyvale, CA4 hours ago $186.9k - $267.7k
...team builds and supports networking products leveraging... ...production support, and engineering collaboration, serving... ...datacenter and cloud networking stacks, establishing... ...data, and leverage AI/ML or LLM-based approaches... ...issues, with the ability to lead technically during...NetworkFull timeTemporary workWork experience placementLocal areaFlexible hoursCISCO Systems
San Jose, CA4 days ago- ...enabling AI breakthroughs, or bringing leading edge products to market, every role... ....THE ROLEWe're seeking a T-shaped engineer ready to elevate their impact.You're... ...architectures including CPU, GPU, FPGA, AI/ML accelerators, and networking devices.Background spanning both...Network
AMD
San Jose, CA3 days ago - ...you can push the limits of what's possible. As a Lead Software Engineer at JPMorganChase within AI/ML Data Platforms, you are an integral part of an... ...systems (CPU/GPU utilization, memory, I/O throughput, networking, scheduling). Experience with Kubernetes fundamentals...Network
JPMorgan Chase & Co.
Palo Alto, CA4 hours ago - ...top-notch technology products. As a Lead Software Engineer at JPMorgan Chase within the Commercial... ...development experience, with emphasis on ML systems. Strong proficiency in... ...or C++. Experience with graph neural networks and graph processing frameworks (DGL, PyTorch...Network
Testing Site - JPMC
Palo Alto, CA4 days ago - ...Elevate your engineering prowess to unprecedented levels by joining a team... ...automation. As a Senior Lead Site Reliability Engineer at... ...excellence with cutting-edge AI/ML capabilities to build... ...community, and continues to expand network and leads evaluation sessions...Network
Testing Site - JPMC
Palo Alto, CA4 days ago $164k - $295k
...you. About The Role: We are looking for a talented Full Stack Engineer to join our growing Production Systems Automation &... ...opportunities for all employees regardless of level or role Employee Networks, geographic neighborhood groups, and volunteer opportunities...NetworkPermanent employmentWork experience placementWork at officeLocal areaFlexible hoursSunnyvale, CA18 days ago$200k - $299k
...Cerebras to deliver industry-leading training and inference speeds... ...RoleAs a lead front-end design engineer, you will be a key part of... ...integration.Demonstrated experience in networking, high-performance computing,... ...high speed IO.Networking stack experience including TCP/IP,...NetworkCerebras Systems
Sunnyvale, CA4 days ago$75k - $85k
...are: We are America’s leading health solutions... ...ambitious Data Scientists and Engineers. In this role, your... ...code and become a full-stack automation engineer?... ...Security, Automation, Network, Cloud, and Artificial... ...of experience with AI/ML, GenAI, and agent development...NetworkHourly payFull timeTemporary workLocal areaRemote workWork from homeShift workCVS Health
San Jose, CA2 days ago$184k - $287.5k
...seeking elite ASIC Infrastructure engineers to deliver the tooling and... ...and RTL Design for the world's leading GPUs. This position offers the... ...by all HW Design teams Use ML/DL/AI techniques to automate Infrastructure... ...compute farm, filer, and network topology requirements at cloud...NetworkFull timeWork experience placementWork at officeWorldwideNvidia
Santa Clara, CA3 days ago- ...Amazon AWS Development Center is seeking a Senior Software Development Engineer for a new security service. The role focuses on designing and delivering a greenfield AI-driven security product, leading teams, and mentoring engineers to produce high-quality, scalable...
Amazon
Santa Clara, CA5 days ago - ...global team of more than 3,000 engineers works across electrical,... ...design disciplines and many more. Leading brands in enterprise... ...characterization plans for complex compute, networking, storage, and high-speed... ..., networking, storage, AI/ML, or other high-performance computing...NetworkLocal area
Celestica
San Jose, CA4 days ago - ...requirement – 2Core Technical Skills:Full-Stack Development: JavaScript/TypeScript, React... ...Angular, Node.js, REST APIs, SQL/NoSQLAI/ML: Python, TensorFlow/PyTorch, Scikit-learn... ...GPUs/TPUs, CPU/memory/storage, edge/IoT, networking)Experience:• Minimum 12+ years of...NetworkContract workH1bRemote work2 days per week
Software Technology Inc
San Jose, CA4 days ago $167.1k - $250.7k
...Technologies, Inc. Job Area: Engineering Group, Engineering Group CPU... ...We are seeking a Senior Full Stack Software Engineer to design, build... ...tools. Integrate AI/ML and LLM technologies into internal... ...Directory. Experience leading technical projects and mentoring...Work experience placementWork from homeQualcomm
Santa Clara, CA4 hours ago$262k - $364k
...above requirements and lead a multi-year technical... ..., racks, space, power, network, cooling to ensure industry... ..., innovation, growth, engineering excellence, landings,... ...problems across the full-stack as we continue to push... ...for deploying Google’s ML hardware. You will help...NetworkTemporary workWorldwideGoogle
Sunnyvale, CA2 days ago- ...Elevate your engineering prowess to unprecedented levels by joining a team... ...reliability. As a Senior Lead Site Reliability Engineer at JPMorgan... ..., and continues to expand network and leads evaluation sessions... ...) Familiarity with AI/ML model building, deployment, and...Network
JPMorgan Chase & Co.
Palo Alto, CA3 days ago $175k - $200k
...to expand its internal AI capabilities, we are seeking a Full Stack Engineer to own the engineering lifecycle of all GenAI-powered applications... ...experience, with at least 2 years working directly with AI/ML APIs or LLM-based applications in a production environment. ~...Full timeRelocationKnightscope
Sunnyvale, CAmore than 2 months ago$140k - $275k
...speed, balancing rapid iteration with strong engineering discipline. Collaborate cross‑... ...Troubleshoot and debug issues across the stack: frontend rendering, backend integration,... ...user‑facing interfaces for technical or AI/ML‑driven products. Knowledge of deployment...Full timeWork at officeShift work$164k - $295k
...CrowdStrike is seeking an experienced Engineering Manager to lead a cloud backend team within our AI Detection... ...products Deep understanding of AI/ML security challenges, threats, and... ...regardless of level or role Employee Networks, geographic neighborhood groups, and...NetworkPermanent employmentWork experience placementWork at officeLocal areaWorldwide2 days per week3 days per weekSunnyvale, CA18 days ago$117k - $234k
...passionate Senior, Software Engineer (Backend) for our team.... ...platform. Our tech stack includes Java for... ...components, multithreading, and networking.Extensive knowledge of... ...with generated AI/ML technologies.Experience... ...professionals within the world's leading retailer who make an...NetworkFull timeTemporary workPart timeWork at officeWalmart
Sunnyvale, CA2 days ago$224k - $356.5k
...world.We are looking for a Senior Software Engineer with deep backend engineering experience... ...exploration, configuration, and marketing workflows.Lead architecture decisions, align partner... ...development. Prior experience in AI or ML is not essential.Ways to stand out from...Full timeShift workNvidia
Santa Clara, CA4 days ago$240k - $333k
...the hardware/software stack, ensuring unparalleled... ...training and serving of large ML models.Act as the... ...bandwidth, and inter-chip network requirements.Manage... ...partnerships across hardware engineering, compiler development,... ....Experience as a lead architect driving multi...NetworkWorldwideShift workGoogle
Sunnyvale, CA3 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Network Stack Lead Engineer. Be the first to apply!
Related searches
- senior ml engineer Cupertino, CA
- data scientist machine learning engineer Cupertino, CA
- machine learning ai engineer Cupertino, CA
- machine learning software engineer Cupertino, CA
- ai ml engineer Cupertino, CA
- machine learning engineer Cupertino, CA
- computer vision machine learning engineer Cupertino, CA
- junior machine learning research engineer Cupertino, CA
- data center network engineer Cupertino, CA
- network engineer level Cupertino, CA