sradc / SmallPebbleLinks
Small deep learning library written from scratch in Python, using NumPy/CuPy.
☆124Updated 3 years ago
Alternatives and similar repositories for SmallPebble
Users that are interested in SmallPebble are comparing it to the libraries listed below
Sorting:
- Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory☆83Updated 2 weeks ago
- Module 0 - Fundamentals☆108Updated last year
- A declarative drawing API in Python☆297Updated last year
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 4 years ago
- Lightweight Experiment & Resource Monitoring 📺☆188Updated 2 years ago
- An experiment with "freely" wired neural networks (no layers)☆164Updated last year
- Annotate python source code☆69Updated 5 years ago
- A simple transformer implementation without difficult syntax and extra bells and whistles.☆53Updated 2 years ago
- Publication-quality data representation library based on Matplotlib.☆167Updated 2 years ago
- A "build to learn" Alpha Zero implementation using Gradient Boosted Decision Trees (LightGBM)☆86Updated 7 months ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention☆56Updated 3 years ago
- ☆40Updated 2 years ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆254Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆242Updated 2 years ago
- A minimal Python kernel so you can run Python in your Python☆39Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- coding an autograd from scratch☆179Updated 6 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆28Updated 2 years ago
- Functional machine learning for fun