Senior Machine Learning Applications and Compiler Engineer, LPX
Core
Develop algorithms and optimizations for NVIDIA's LPX inference and compiler stack, mapping neural network workloads onto future NVIDIA platforms.
Role type
Senior IC machine learning applications and compiler engineer
Builds
High-performance runtime and compiler components for end-to-end inference optimization
Domain
Deep learning systems, compiler technology, and spatial processor architectures
Deliverable
production ML models
Required skills
Systems-level programming (C/C++/Rust), compiler/runtime development (IR design, optimization passes, code generation), LLVM/MLIR expertise, deep learning frameworks (TensorFlow/PyTorch/ONNX), parallel/heterogeneous compute architectures, profiling and benchmarking
Preferred skills
Spatial/dataflow architectures (static scheduling, pipeline/tensor parallelism), open-source contributions to ML frameworks/compilers, research publications at top-tier venues, large-scale AI distributed inference systems
Technologies
LLVM, MLIR, TensorFlow, PyTorch, ONNX, C, C++, Rust
