Tensors and Dynamic neural networks in Python with strong GPU acceleration
Repository Index
47,133 repositories indexed
1–10 of 10 · Most stars
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Open Machine Learning Compiler Framework
NumPy & SciPy for GPU
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
High-performance TensorFlow library for quantitative finance.
NVIDIA cuML: GPU-Accelerated Machine Learning
DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome
🧱 Experimental library for building inference engines