lly-zero-one Goto Github PK

aitemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

caffe2-benchmarking

Benchmark Caffe2 performance and accuracy

classyvision

An end-to-end PyTorch framework for image and video classification

detectron2

Detectron2 is FAIR's next-generation research platform for object detection and segmentation.

dlrm

An implementation of a deep learning recommendation model (DLRM)

fairscale

PyTorch extensions for high performance and large scale training.

fbgemm

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

hugectr

HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

nebullvm

Plug and play modules to optimize the performances of your AI systems 🚀

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

ray

An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.

robustvideomatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

semanticgan_code

Official repo for SemanticGAN https://nv-tlabs.github.io/semanticGAN/

sparse-ads-baselines

trtorch

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

tvm-2

TVM integration into PyTorch