Senior Machine Learning Applications and Compiler Engineer, LPX
Tasks
- Benchmark profile and monitor performance metrics
- Build runtime components for inference optimization
- Collaborate with hardware architects on software observations
- Define workload mappings onto NVIDIA systems
- Develop compiler components for neural network workloads
- Implement graph transformations scheduling and memory optimizations
- Integrate with NVIDIA software ecosystem for model deployment
- Optimize compiler mappings of neural network graphs
- Prototype compilation runtime techniques
- Publish technical work on compilation approaches
Perks/Benefits
- N/A
Skills/Tech-stack
Benchmarking | C++ | Code generation | Debugging | GPU Computing | Graph optimization | Heterogeneous computing | IR design | LLVM | LLVM IR | MLIR | MLIR passes | ONNX | Parallel Computing | Profiling | PyTorch | Rust | TensorFlow
Education
Regions
Countries
States
Related jobs
-
Senior Platform AI Engineer USD 184K-356KAuthentication | Authorization | C# | C++ | CachingComprehensive benefitsSenior-level Full TimeUS, CA, Santa Clara R1d ago
-
Senior-level Full TimeUS, CA, Santa Clara R2d ago
-
C++ | Caching | Debugging | Distributed Systems | GPU memorySenior-level Full TimeUS, CA, Santa Clara R2d ago
-
Senior Software Engineer, Agentic Engineering USD 184K-356KAgent Orchestration | Autonomous Search | CI/CD | Code generation | Compiler infrastructureEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R8d ago
-
Senior Storage Software Engineer, DGXC Data Services USD 152K-287KAPI Design | Algorithms | C++ | Caching | ConcurrencySenior-level Full TimeUS, CA, Santa Clara R9d ago
-
Senior Software Engineer - AI Research Clusters USD 152K-287KAIOps | Agentic AI | C++ | CSS | Database ModelingEquity | Health insurance | Paid time off | Retirement planSenior-level Full TimeUS, CA, Santa Clara R9d ago
-
Software Engineer - AI Research Clusters USD 124K-195KAIOps | Agentic AI | Automated Deployment | C++ | CSSBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R9d ago
-
AI Software Engineer USD 168K-322KAPI Development | Algorithms | Analytics | Automated reporting | Data IngestionEquity | Health benefits | Paid time offSenior-level Full TimeUS, CA, Santa Clara R9d ago
-
Senior Software Engineer, PyTorch - Deep Learning USD 184K-356KC++ | CUDA | Deep Learning Compilers | Deep learning | Distributed Parallel ProgrammingSenior-level Full TimeUS, CA, Santa Clara R12d ago
-
C++ | Cache Management | Distributed Systems | GPU memory | GPU memory managementAgile team environment | Equity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R13d ago
-
Senior Software Engineer - Storage USD 152K-287KAmazon Web Services | Automation | Azure | BeeGFS | C++Senior-level Full TimeUS, CA, Santa Clara R13d ago
-
Senior Software Engineer, DGXC Data Services USD 152K-287KAWS | Algorithms | Apache Iceberg | Apache Spark | AzureEmployee benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R20d ago
-
Senior Software Engineer, AI Storage USD 184K-287KAlgorithms | Bash | C++ | CUDA | CloudBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R23d ago
-
Senior Deep Learning Framework Communications Engineer USD 152K-287KC++ | CUDA | CUDA kernels | CuTe | Distributed TrainingBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R25d ago
-
C++ | CUDA | Docker | Infiniband | JAXSenior-level Full TimeUS, CA, Santa Clara R26d ago
-
Senior Deep Learning Frameworks CUDA Software Engineer USD 184K-356KAutograd | C++ | CUDA | Compiler technology | Computer ArchitectureSenior-level Full TimeUS, CA, Santa Clara R29d ago
-
Senior Scientific Machine Learning Engineer – Earth-2 USD 152K-287KCUDA | Containers | Data parallelism | Diffusion Models | GPU KernelBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R1mo ago
-
Senior Storage Production Engineer - DGX Cloud USD 176K-333KAI/ML | Access Control | Algorithms | Ansible | AuditingBenefits | Equity | On-call rotationSenior-level Full TimeUS, CA, Santa Clara R1mo ago