Sglang team at NVIDIA (Deep learning frameworks).
- Irvine, Californa
Stars
A machine learning compiler for GPUs, CPUs, and ML accelerators
CUDA Templates and Python DSLs for High-Performance Linear Algebra
A scalable code framework for CuDNN frontend APIs
A visualization tool to display TF-Grappler optimized op graph
Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Convolution NN resize initialization for subpixel convolutions