Senior Deep Learning Compiler Engineer - XLA
Tasks
- Collaborate with deep learning framework teams
- Collaborate with hardware architecture teams to accelerate next generation AI compilers
- Design graph partitioning and tensor sharding for distributed training and inference
- Design user-facing features in JAX and related libraries
- Develop compiler optimization algorithms for deep learning workloads
- Generate code for NVIDIA GPU backends using MLIR and LLVM
- Implement compiler optimization techniques for neural network graphs
- Optimize inference and training performance for JAX and OpenXLA
- Perform performance tuning and analysis
Perks/Benefits
- N/A
Skills/Tech-stack
C# | C++ | CUDA | Code generation | Debugging | Distributed Computing | Graph partitioning | High Performance | High-Performance Computing | JAX | LLVM | MLIR | OpenAI Triton | OpenCL | OpenXLA | Performance Computing | Performance Tuning | PyTorch | Software testing | TVM | Tensor sharding | TensorFlow | XLA
Education
Regions
Countries
States
Related jobs
-
Senior Agentic AI Software Engineer USD 152K-241KAI Agents | Artificial Intelligence | Evaluation | Inference | Language ProcessingEquity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R3d 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 R3d ago
-
C++ | CUDA | CuDF | DataFrame | Database accelerationSenior-level Full TimeUS, CA, Santa Clara R3d ago
-
C++ | CUDA | Docker | Infiniband | JAXSenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Senior Software Engineer, AI Performance Analysis USD 184K-287KAI | C++ | Integration | Machine Learning | Performance AnalysisEmployee benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Distinguished Engineer, Scaled Out Inferencing USD 320K-488KAI infrastructure | CUDA | Cloud Native | Cloud Native Architecture | Container OrchestrationSenior-level Full TimeUS, CA, Santa Clara R6d ago
-
Software Engineer, CUDA Deep Learning Systems USD 152K-258KC++ | CUDA | Computer Architecture | Deep learning | Distributed ComputingBenefits | EquityMid-level Full TimeUS, CA, Santa Clara R6d ago
-
Machine Learning Engineer, AI Safety USD 152K-281KBias Mitigation | Content Safety | Deep learning | Keras | Language ModelsComprehensive benefits packageMid-level Full TimeUS, CA, Santa Clara R6d ago
-
C++ | CUDA | CUPTI | Continuous batching | KV cacheSenior-level Full TimeUS, CA, Santa Clara R11d ago
-
AWS | Azure | Bazel | C# | C++Equity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R11d ago
-
C# | C++ | Caching | Distributed Systems | GPU ArchitectureComprehensive benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Senior Software Engineer – TensorRT Edge-LLM USD 152K-241KAttention | C++ | CUDA | Compiler optimization | FlashInferSenior-level Full TimeUS, CA, Santa Clara R12d 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 R12d ago
-
Senior Deep Learning Algorithm Engineer USD 152K-241KAutoscaling | C++ | Deep learning | Distributed Systems | Inference OptimizationSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Senior Storage Production Engineer - DGX Cloud USD 184K-333KAccess Control | Anomaly Detection | Ansible | Auditing | BashSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
C++ | CPU architecture | CUDA | Deep learning | Fraud DetectionSenior-level Full TimeUS, CA, Santa Clara R12d 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 R14d ago
-
Senior Deep Learning Frameworks CUDA Software Engineer USD 184K-356KC++ | CUDA | Distributed machine learning | HPC | JAXSenior-level Full TimeUS, CA, Santa Clara R14d ago
-
Senior Deep Learning Framework Communications Engineer USD 184K-287KC++ | CUDA | CuTe | Elasticity | Fault ToleranceBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R14d ago
-
Senior Cloud Software Engineer, DGXC Data Services USD 152K-241KAPI Design | Algorithms | Amazon Web Services | C++ | CheckpointsEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R15d ago
-
C# | C++ | CUDA | Communication Scaling | LinuxEquity | Health insurance | Paid time off | Retirement planSenior-level Full TimeUS, CA, Santa Clara R15d ago
-
Senior Software Engineer, AI Storage USD 184K-287KAlgorithms | Bash | C# | C++ | CUDABenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R15d ago
-
Senior Platform AI Engineer USD 184K-287KAuthentication | Authorization | Backend Infrastructure | C# | C++Senior-level Full TimeUS, CA, Santa Clara R20d ago
-
Asynchronous programming | Context Management | Deep learning | Distributed Systems | GPUSenior-level Full TimeUS, CA, Santa Clara R21d ago