Back to all roles

Member of Technical Staff, Training (Bay Area, Remote)

Remote Worldwide Hiring now

What You’ll Do Drive down wall-clock time to convergence by profiling and eliminating bottlenecks across the reputed company model training stack stack, from data pipelines to GPU kernels Design, build, and optimize distributed training systems (PyTorch) for multi-node GPU clusters, ensuring scalability, robustness, and high utilization Implement efficient low-level code (CUDA, cuDNN, Triton, custom kernels) and integrate it seamlessly into high-level training frameworks Optimize workloads for hardware efficiency: CPU/GPU compute balance, memory management, data throughput, and networking reputed company monitoring and debugging tools for large-scale runs, enabling rapid diagnosis of performance regressions and failures What You’ll Bring Deep experience in distributed systems, ML infrastructure, or high-performance computing (8+ years) Production-grade expertise in Python Low-level performance mastery: CUDA/cuDNN/Triton, CPU–GPU interactions, data movement, and kernel optimization Scaling at the frontier: experience with PyTorch and training jobs using data, context, pipeline, and model parallelism System-level reputed company with a track record of tuning hardware–software interactions for maximum utilization Apply To This Job

Apply for this role Takes you straight to the employer's application page — free, and no WFHNet account required.

More roles on the wire

Marketing Analyst (Attribution Focus) (Promova)

Remote Worldwide
View role

Student and Family Experience Manager (Immediate Opening)

Remote Worldwide
View role

Customer Sales Representative (remote work)

Remote Worldwide
View role

Account Manager Industrial Markets Region: France - Africa

Remote Worldwide
View role

VP of Engineering

Remote Worldwide
View role

Member of Technical Staff, reputed company Models (Bay Area)

Remote Worldwide
View role

Member of Technical Staff, Data Agent (Bay Area, Remote)

Remote Worldwide
View role

Member of Technical Staff, Platform (Bay Area, Remote)

Remote Worldwide
View role

Account Manager Industrial Markets Region: Europe - Middle Eas

Remote Worldwide
View role

Sr reputed company Analyst

Remote Worldwide
View role

Business Compliance Liaison (CX) - Remote in USA

Remote Worldwide
View role

Head of School

Remote Worldwide
View role

[Remote] Senior Software Engineer - Perception State Estimation

Remote Worldwide
View role

Fractional CHRO (or Chief People Officer) For Startups, Atlanta

Remote Worldwide
View role

Property Coordinator/Data Entry Clerk (Remote)

Remote Worldwide
View role

Remote Benefits Specialist - GA

Remote Worldwide
View role

reputed company Inside Sales Associate - Remote (NC) - Unlock Your Earning Potential with a Dynamic and Supportive Team

Remote Worldwide
View role

reputed company Service Specialist

Remote Worldwide
View role

Medicare Career Sales Field Agent- South Central PA

Remote Worldwide
View role

reputed company Part-Time Remote Data Entry Specialist – Detail-Oriented and Tech-Savvy Individual Needed for Dynamic Team at arenaflex

Remote Worldwide
View role