FROG: Fast Real-time Optimization of Gradients
Project description
frog ![unit test badge](https://pypi-camo.freetls.fastly.net/c79af04210eabcb40b90b5285895a99966c104ef/68747470733a2f2f6769746875622e636f6d2f6b65766275682f66726f672f616374696f6e732f776f726b666c6f77732f746573742e796d6c2f62616467652e737667)
frog: fast real-time optimization of gradients
a beautifully compact machine-learning library
homepage | documentation | examples | pip
Installation
pip install froog
Overview of Features
- Tensors
- Automatic Differentiation
- Forward and backward passes
- Input/gradient shape-tracking
- MNIST example
- 2D Convolutions (im2col)
- Numerical gradient checking
- The most common optimizers (SGD, Adam, RMSProp)
Math Operations
- Scalar-Matrix Multiplication
- Dot Product
- Sum
- ReLU
- Log Softmax
- 2D Convolutions
- Avg & Max pooling
- More
Bounties
Want to help but don't know where to start?
Our top bounty is to get EfficientNet v2 model working inside of the examples folder.
Easy
- built in MLP model
- binary cross entropy
- dropout layer
- flatten
Medium
- simplify how context and gradients are handled
Hard
- efficientNet
- transformers
- stable Diffusion
- winograd Convs
- MPS support
- CUDA support
Contributing
here are some basic guidelines for contributing:
- reduce complexity (currently at 585 lines of code)
- increase speed
- add features, must include tests
- in that order
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.1.5.tar.gz
(10.5 kB
view hashes)
Built Distribution
froog-0.1.5-py3-none-any.whl
(10.5 kB
view hashes)