Senior Deep Learning Framework Communications Engineer
Tasks
- Analyze multi GPU communication requirements in AI workloads
- Author custom communication or fused compute communication kernels
- Characterize AI workload performance on multi GPU clusters
- Design fault tolerant elastic solutions for large scale AI workloads
- Develop and influence communication library roadmap
- Improve AI compilers to optimize communication
- Integrate communication libraries features into AI frameworks
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | CUDA | Communication overlap | Compiler optimization | Compute Communication Overlap | CuTe | Distributed Training | Distributed inference | Elasticity | Fault Tolerance | GPUDirect | HPC communication | JAX | MPI | Mixture of Experts | NCCL | NVIDIA Nsight | NVSHMEM | Nsight Systems | Nvidia NSight Systems | Parallel Programming | Performance Benchmarking | PyTorch | PyTorch Profiler | Python | Reinforcement Learning | SGLang | TRT-LLM | Topology Discovery | Torch compile | Triton | VLLM
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
Regions
Countries
States
Related jobs
-
C# | C++ | DRAM | Distributed Systems | GPUSenior-level Full TimeUS, CA, Santa Clara R1d ago
-
Principal Deep Learning Algorithm Engineer USD 272K-431KC++ | CUDA | Computer Architecture | Deep learning | GPU ProgrammingBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Senior Cloud Software Engineer, DGXC Data Services USD 152K-241KAPI Design | AWS | Algorithms | C++ | Data StructuresBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R3d ago
-
C++ | CUDA | Containerization | Distributed Systems | DockerSenior-level Full TimeUS, CA, Santa Clara R4d ago
-
C++ | Computer Vision | Curriculum learning | Data pipeline | Deep learningSenior-level Full TimeUS, CA, Santa Clara R5d ago
-
C++ | CUDA | Distributed inference | GPU Architecture | Graph CompilerBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Senior System Software Engineer – Dynamo Tools USD 152K-241KAWS | Automation | Azure | Cloud platform | ConcurrencyComprehensive benefits package | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Senior Software Engineer, AI Storage USD 184K-287KAlgorithms | Bash | C# | C++ | CUDABenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Bash | Bootstrap | CSI | CSS3 | Container StorageSenior-level Full TimeUS, CA, Santa Clara R6d ago
-
Senior Storage Production Engineer - DGX Cloud USD 176K-333KAccess Control | Alerting | Ansible | Auditing | BackupBenefits | Equity | On-call rotation supportSenior-level Full TimeUS, CA, Santa Clara R8d ago
-
C++ | CUDA | Computer Vision | Deep learning | Generative AIComprehensive benefits | Equity | Family benefitsSenior-level Full TimeUS, CA, Santa Clara R9d ago
-
Senior-level Full TimeUS, CA, Santa Clara R11d ago
-
Applied Agentic AI Lead, Partner Co-Design USD 224K-356KAgent systems | Compliance governance | Deep learning | Docker | Efficient Fine TuningEquity | Health benefits | Professional developmentSenior-level Full TimeUS, CA, Santa Clara R11d ago
-
C++ | Cache Management | Debugging | Distributed Systems | GPU memorySenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Algorithms | Benchmarking | C++ | Capacity Planning | Code generationEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R12d ago