dblalock / bolt
10x faster matrix and vector operations
☆2,480Updated 2 years ago
Alternatives and similar repositories for bolt:
Users that are interested in bolt are comparing it to the libraries listed below
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,029Updated 10 months ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,963Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,410Updated this week
- JAX-based neural network library☆2,960Updated 3 weeks ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,342Updated this week
- Library for reading and writing large multi-dimensional arrays.☆1,381Updated last week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,165Updated this week
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,556Updated last year
- PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.☆756Updated last year
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,529Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,598Updated 2 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,261Updated last week
- PyTorch extensions for high performance and large scale training.☆3,260Updated last month
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,677Updated this week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,895Updated 8 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,741Updated last week
- Up to 200x Faster Dot Products & Similarity Metrics — for Python, Rust, C, JS, and Swift, supporting f64, f32, f16 real & complex, i8, an…☆1,238Updated this week
- maximal update parametrization (µP)☆1,447Updated 7 months ago
- Lightning fast C++/CUDA neural network framework☆3,879Updated 3 weeks ago
- common in-memory tensor structure☆942Updated last week
- Optax is a gradient processing and optimization library for JAX.☆1,802Updated this week
- Efficiently computes derivatives of NumPy code.☆7,138Updated this week
- AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.☆2,221Updated this week
- CUDA Templates for Linear Algebra Subroutines☆6,233Updated last week
- High-performance automatic differentiation of LLVM and MLIR.☆1,337Updated this week
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,533Updated this week
- RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…☆838Updated this week
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,822Updated last month
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,417Updated 6 months ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,396Updated this week