Tensors and Dynamic neural networks in Python with strong GPU acceleration
- neural-network
- autograd
- gpu
- numpy
47,133 repositories indexed
1–22 of 22 · Most stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
AirLLM 2.8T LLM inference with single 4GB GPU
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
The fastai deep learning library
A high-performance, zero-overhead, extensible Python compiler with built-in NumPy support
Open3D: A Modern Library for 3D Data Processing
Solve puzzles. Learn CUDA.
NumPy & SciPy for GPU
cuDF - GPU DataFrame Library
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.
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.
Bringing stable diffusion models to web browsers. Everything runs inside the browser with no server support.
Up to 100x faster strings for C, C++, CUDA, Python, Rust, Swift, JS, & Go, leveraging NEON, AVX2, AVX-512, SVE, GPGPU, & SWAR to accelerate search, hashing, sorting, edit distances, sketches, and memory ops 🦖
🛠 All-in-one web-based IDE specialized for machine learning and data science.
Beautiful, open source, WebGPU-based charting library
PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
ProtTrans is providing state of the art pretrained language models for proteins. ProtTrans was trained on thousands of GPUs from Summit and hundreds of Google TPUs using Transformers Models.
⚡ Datoviz: high-performance GPU rendering for scientific data visualization
High-performance Bayesian Data Analysis on the GPU in Clojure
Open source math and physics engine that executes MATLAB-syntax code. Features a modern compiler, automatic GPU acceleration for large calculations, and support for math on geometry / FEA calculations.
Go's answer to pandas + scikit-learn: a batteries-included data analysis library with dataframes, statistics, machine learning, quant, and GPU acceleration. Run Python with seamless two-way variable sharing, or explore data straight from the Insyra CLI.