Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
May 18, 2024 - Python
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Efficient CPU/GPU/Vulkan ML Runtimes for VapourSynth (with built-in support for waifu2x, DPIR, RealESRGANv2/v3, Real-CUGAN, RIFE, SCUNet and more!)
On-device AI across mobile, embedded and edge for PyTorch
Neural Networks with Sparse Weights in Rust using GPUs, CPUs, and FPGAs via CUDA, OpenCL, and oneAPI
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.
SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance
SkyPilot: Run LLMs, AI, and Batch jobs on any cloud. Get maximum savings, highest GPU availability, and managed execution—all with a simple interface.
Open deep learning compiler stack for cpu, gpu and specialized accelerators
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos
GEOS Simulation Framework
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
Video stabilization using gyroscope data
Add a description, image, and links to the gpu topic page so that developers can more easily learn about it.
To associate your repository with the gpu topic, visit your repo's landing page and select "manage topics."