a beautifully simplistic ml framework
Project description
ribbit
ribbit: fast real-time optimization of gradients
a beautifully compact machine-learning library
homepage | documentation | pip
RIBBIT is a SUPER SIMPLE machine learning framework with the goal of creating tools with AI --> easily and efficiently.
It's an end-to-end framework, encapsulating everything from ensemble trees to deep neural networks (still working on all that lol)
Installation
pip install ribbit
Overview of Features
- Custom Tensors
- Backpropagation
- Automatic Differentiation (autograd)
- Forward and backward passes
- ML Operations
- 2D Convolutions (im2col)
- Numerical gradient checking
- Acceleration methods (Adam)
- Avg & Max pooling
- EfficientNet inference
- GPU Support
- and a bunch more
Sneak Peek
from ribbit.tensor import Tensor
from ribbit.utils import Linear
import ribbit.optim as optim
class mnistMLP:
def __init__(self):
self.l1 = Tensor(Linear(784, 128))
self.l2 = Tensor(Linear(128, 10))
def forward(self, x):
return x.dot(self.l1).relu().dot(self.l2).logsoftmax()
model = mnistMLP()
optim = optim.SGD([model.l1, model.l2], lr=0.001)
Bounties
THERES LOT OF STUFF TO WORK ON! VISIT THE BOUNTY SHOP
Pull requests will be merged if they:
- increase simplicity
- increase functionality
- increase efficiency
more info on contributing
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
froog-0.2.4.tar.gz
(19.4 kB
view hashes)
Built Distribution
froog-0.2.4-py3-none-any.whl
(17.4 kB
view hashes)