Senior Deep Learning Framework Communications Engineer
Tasks
- Author custom communication or fused compute communication kernels
- Benchmark AI cluster performance
- Conduct AI workload performance characterization
- Design fault tolerant elastic solutions
- Improve AI compilers for communication hiding
- Influence roadmaps for communication libraries
- Integrate communication libraries features into AI frameworks
- Perform multi GPU communication requirement analysis
Perks/Benefits
Skills/Tech-stack
C++ | CUDA | CuTe | Elasticity | Fault Tolerance | GPUDirect | JAX | MPI | NCCL | NVIDIA Nsight | NVSHMEM | Nsight Systems | Nvidia NSight Systems | Performance Benchmarking | PyTorch | PyTorch Profiler | Python | SGLang | TRT-LLM | Topology Discovery | Torch.compile | Triton | VLLM
Education
Regions
Countries
States
Related jobs
-
Software Engineer (Applied AI/ML) USD 200K-250KCI/CD | Data Pipelines | Data ontologies | Design of Experiments | Digital Twins401k | Bonuses | Equity | FSA | Flexible time offMid-level Full TimeSanta Clara, CA or Remote R5d ago
-
Senior Solutions Architect, Inference Service Providers USD 184K-287KAgent Design | Cache Management | Custom Inference Kernels | Data Curation | DevOpsSenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Array computing | Asynchronous programming | C++ | CUDA | Compiler optimizationBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Senior Software Engineer, RL Post-Training Frameworks USD 184K-287KActor Based Programming | C# | C++ | Deep learning | Distributed SystemsSenior-level Full TimeUS, CA, Santa Clara R5d ago
-
AWS | Azure | C# | C++ | Computer VisionEntry-level InternshipUS,CA,Santa Clara R5d ago
-
AWS | Azure | C# | C++ | Computer VisionHybrid or onsite workEntry-level InternshipUS,CA,Santa Clara R5d ago
-
Senior-level Full TimeUS, CA, Santa Clara R6d ago
-
Senior AI Infrastructure Software Engineer - DGX Cloud USD 184K-287KC# | C++ | Cause analysis | Debugging | Distributed SystemsBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R6d ago
-
Senior-level Full TimeUS, CA, Santa Clara R7d ago
-
BEV | C++ | Computer Vision | Data Augmentation | Data SimulationBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R8d ago
-
Senior Deep Learning Algorithm Engineer USD 184K-287KCPU architecture | Distributed Computing | GPU Architecture | Generative AI | JAXEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Senior Deep Learning Compiler Engineer - XLA USD 152K-241KC# | C++ | CUDA | Code generation | DebuggingSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
C++ | CUDA | CuDF | DataFrame | Database accelerationSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
C++ | CUDA | Docker | Infiniband | JAXSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Distinguished Engineer, Scaled Out Inferencing USD 320K-488KAI infrastructure | CUDA | Cloud Native | Cloud Native Architecture | Container OrchestrationSenior-level Full TimeUS, CA, Santa Clara R15d ago
-
Software Engineer, CUDA Deep Learning Systems USD 152K-258KC++ | CUDA | Computer Architecture | Deep learning | Distributed ComputingBenefits | EquityMid-level Full TimeUS, CA, Santa Clara R15d ago
-
AWS | Azure | Bazel | C# | C++Equity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R20d ago
-
C# | C++ | Caching | Distributed Systems | GPU ArchitectureComprehensive benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | Computer Architecture | Deep Learning Compilers | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Senior Deep Learning Algorithm Engineer USD 152K-241KAutoscaling | C++ | Deep learning | Distributed Systems | Inference OptimizationSenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Senior Storage Production Engineer - DGX Cloud USD 184K-333KAccess Control | Anomaly Detection | Ansible | Auditing | BashSenior-level Full TimeUS, CA, Santa Clara R21d ago
-
C++ | CPU architecture | CUDA | Deep learning | Fraud DetectionSenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Bash | Bootstrap | CSI | CSS3 | Container StorageSenior-level Full TimeUS, CA, Santa Clara R22d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | CUDA kernels | Computer Architecture | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R23d ago
-
Senior Deep Learning Frameworks CUDA Software Engineer USD 184K-356KC++ | CUDA | Distributed machine learning | HPC | JAXSenior-level Full TimeUS, CA, Santa Clara R23d ago