sradc / SmallPebble
Minimal deep learning library written from scratch in Python, using NumPy/CuPy.
☆121Updated 2 years ago
Alternatives and similar repositories for SmallPebble:
Users that are interested in SmallPebble are comparing it to the libraries listed below
- Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory☆80Updated last week
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- An experiment with "freely" wired neural networks (no layers)☆163Updated 7 months ago
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- A declarative drawing API in Python☆294Updated 6 months ago
- A recurrent/memory-based neural network library implemented from scratch in C.☆56Updated 3 years ago
- ☆242Updated 11 months ago
- ☆18Updated 2 years ago
- Autograd Engine for Python Written in Python C-API for Speed☆20Updated last year
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 3 years ago
- Publication-quality data representation library based on Matplotlib.☆167Updated 2 years ago
- coding an autograd from scratch☆172Updated 6 years ago
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆106Updated 2 months ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆250Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆236Updated last year
- Python library for embedding large graphs in 2D space, using force-directed layouts.☆163Updated 2 years ago
- A lightweight deep learning framework☆32Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Lightweight Experiment & Resource Monitoring 📺☆186Updated 2 years ago
- Clover: Quantized 4-bit Linear Algebra Library☆112Updated 6 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆91Updated 3 months ago
- Annotate python source code☆69Updated 4 years ago
- ☆85Updated 4 years ago
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆317Updated 5 months ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- ☆131Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆85Updated last year
- ☆30Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago