sradc / SmallPebbleLinks
Small deep learning library written from scratch in Python, using NumPy/CuPy.
☆123Updated 2 years ago
Alternatives and similar repositories for SmallPebble
Users that are interested in SmallPebble are comparing it to the libraries listed below
Sorting:
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- A declarative drawing API in Python☆295Updated 9 months ago
- An experiment with "freely" wired neural networks (no layers)☆163Updated 10 months ago
- Annotate python source code☆69Updated 5 years ago
- Publication-quality data representation library based on Matplotlib.☆167Updated 2 years ago
- ☆243Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 3 weeks ago
- Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory☆80Updated last month
- A "build to learn" Alpha Zero implementation using Gradient Boosted Decision Trees (LightGBM)☆86Updated 2 months ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Lightweight Experiment & Resource Monitoring 📺☆187Updated 2 years ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆251Updated last year
- Docs☆143Updated 6 months ago
- Named tensors with first-class dimensions for PyTorch☆331Updated last year
- convert a scikit-learn decision tree into a Keras model☆39Updated last year
- coding an autograd from scratch☆177Updated 6 years ago
- Execute a jupyter notebook, fast, without needing jupyter☆131Updated last month
- A minimal Python kernel so you can run Python in your Python☆39Updated 3 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention☆56Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆239Updated 2 years ago
- A lightweight deep learning framework☆34Updated 4 years ago
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆113Updated last year
- Lightweight machine learning library based on OpenCL 1.2☆73Updated 4 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 5 months ago
- Reference implementation of "An Algorithm for Routing Vectors in Sequences" (Heinsen, 2022) and "An Algorithm for Routing Capsules in All…☆169Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- ☆134Updated 2 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Memory mapped numpy arrays of varying shapes☆298Updated 11 months ago