sradc / SmallPebble
Minimal deep learning library written from scratch in Python, using NumPy/CuPy.
☆120Updated 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☆81Updated last week
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- Publication-quality data representation library based on Matplotlib.☆167Updated 2 years ago
- Module 0 - Fundamentals☆101Updated 5 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last week
- Lightweight Experiment & Resource Monitoring 📺☆185Updated last year
- MinT: Minimal Transformer Library and Tutorials☆252Updated 2 years ago
- A "build to learn" Alpha Zero implementation using Gradient Boosted Decision Trees (LightGBM)☆86Updated 4 years ago
- arxiv_miner is a toolkit for mining research papers on CS ArXiv.☆130Updated 10 months ago
- A declarative drawing API in Python☆294Updated 5 months ago
- ☆18Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention☆57Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Curated list of awesome material on optimization techniques to make artificial intelligence faster and more efficient 🚀☆113Updated last year
- Annotate python source code☆69Updated 4 years ago
- Autograd Engine for Python Written in Python C-API for Speed☆20Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆89Updated 2 months ago
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆315Updated 4 months ago
- An experiment with "freely" wired neural networks (no layers)☆163Updated 6 months ago
- A High Level API for Deep Learning in JAX☆472Updated 2 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 last month
- An open-source efficient deep learning framework/compiler, written in python.☆681Updated last week
- coding an autograd from scratch☆172Updated 6 years ago
- Execute a jupyter notebook, fast, without needing jupyter☆127Updated last month
- An implementation of convolutional networks in NumPy!☆80Updated 3 years ago
- 1.2% test error on MNIST using only least squares and numpy calls.☆17Updated last year
- A deep learning framework created from scratch with Python and NumPy☆237Updated 2 years ago
- Memory mapped numpy arrays of varying shapes☆292Updated 8 months ago