Tensors and Dynamic neural networks in Python with strong GPU acceleration
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
Distributed and decentralized training framework for PyTorch over graph
A Minimalistic Auto-Diff Optimization Framework for Teaching and Understanding Pytorch
(yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.
This repository aims at setting up a Nvidia's Triton Inference server which simplifies the deployment of AI models at scale in production. It natively supports multiple framework backends like TensorFlow, PyTorch, ONNX Runtime, Python, and even custom backends.