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:
- A "build to learn" Alpha Zero implementation using Gradient Boosted Decision Trees (LightGBM)☆86Updated 9 months ago
- Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory☆83Updated 2 weeks ago
- Module 0 - Fundamentals☆111Updated last year
- Publication-quality data representation library based on Matplotlib.☆168Updated 3 years ago
- A minimal implementation of autograd (in pure Python) 🍰☆96Updated 4 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆149Updated 2 years ago
- Lightweight Experiment & Resource Monitoring 📺☆187Updated 2 years ago
- Annotate python source code☆69Updated 5 years ago
- A simple transformer implementation without difficult syntax and extra bells and whistles.☆54Updated 2 years ago
- An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention☆56Updated 3 years ago
- convert a scikit-learn decision tree into a Keras model☆39Updated 2 years ago
- A declarative drawing API in Python☆298Updated last year
- Curated list of awesome material on optimization techniques to make artificial intelligence faster and more efficient 🚀☆119Updated 2 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆28Updated 2 years ago
- A minimal Python kernel so you can run Python in your Python☆39Updated 3 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 …☆107Updated 3 weeks ago
- A lightweight (serverless) native python parallel processing framework based on simple decorators and call graphs.☆103Updated 3 years ago
- ☆40Updated 3 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆242Updated 2 years ago
- Celular automaton-based calculus for the masses☆111Updated 5 years ago
- A pure NumPy implementation of Mamba.☆222Updated last year
- An Implementation of Transformer in Transformer in TensorFlow for image classification, attention inside local patches☆43Updated 3 years ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆254Updated 2 years ago
- Run compute jobs on AWS as if you were running them locally.☆124Updated 4 years ago
- Execute a jupyter notebook, fast, without needing jupyter☆137Updated last month
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- ☆249Updated last year
- Visualize tensors in a plain Python REPL using Sparklines☆45Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago