Tensors and Dynamic neural networks in Python with strong GPU acceleration
- neural-network
- autograd
- gpu
- numpy
Repository Index
47,133 repositories indexed
1–24 of 91 · Most stars
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
The fastai deep learning library
High-performance In-browser LLM Inference Engine
A WebGL accelerated JavaScript library for training and deploying ML models.
- deep-learning
- typescript
- webgl
- javascript
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
- autodiff
- deep-learning
- machine-learning
- rust
🤖 GPU accelerated Neural networks in JavaScript for Browsers and Node.js
- neural-network
- brain
- recurrent-neural-networks
- easy-to-use
Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...
- java
- gpu
- deeplearning
- neural-nets
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
- tensorrt
- nvidia
- deep-learning
- inference
Solve puzzles. Learn CUDA.
- cuda
- machine-learning
- puzzles
NumPy & SciPy for GPU
- cuda
- cudnn
- cublas
- cusolver
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.
- cloud-computing
- deep-learning
- gpu
- hyperparameter-tuning
High-performance GPGPU inference of OpenAI's Whisper automatic speech recognition (ASR) model
CUDA Templates and Python DSLs for High-Performance Linear Algebra
- cuda
- deep-learning
- deep-learning-library
- cpp
cuDF - GPU DataFrame Library
- gpu
- rapids
- cudf
- arrow
A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.
- machine-learning
- decision-trees
- gradient-boosting
- gbm
H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.
High-performance TensorFlow library for quantitative finance.
NVIDIA cuML: GPU-Accelerated Machine Learning
- machine-learning-algorithms
- machine-learning
- cuda
- gpu
Optimized primitives for collective multi-GPU communication
- communications
- cpp
- cuda
- deep-learning
Run Keras models in the browser, with GPU support using WebGL