froog 0.2.8__tar.gz → 0.3.1__tar.gz

froog-0.3.1/PKG-INFO

@@ -0,0 +1,301 @@
+Metadata-Version: 2.1
+Name: froog
+Version: 0.3.1
+Summary: a beautifully simplistic tensor library
+Author: Kevin Buhler
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: requests
+Requires-Dist: matplotlib
+Requires-Dist: urllib
+
+# froog <img src="https://github.com/kevbuh/froog/actions/workflows/test.yml/badge.svg" alt="unit test badge" > <img src="https://static.pepy.tech/badge/froog" alt="num downloads badge">
+<div align="center" >
+  <img src="https://raw.githubusercontent.com/kevbuh/froog/main/assets/froog.png" alt="froog the frog" height="200">
+  <br/>
+  froog: fast real-time optimization of gradients 
+  <br/>
+  a beautifully compact tensor library
+  <br/>
+  <a href="https://github.com/kevbuh/froog">homepage</a> | <a href="https://github.com/kevbuh/froog/tree/main/docs">documentation</a> | <a href="https://pypi.org/project/froog/">pip</a>
+  <br/>
+  <br/>
+</div>
+
+```froog``` is an easy-to-read tensor library (<a href="https://www.pepy.tech/projects/froog">16k pip installs!</a>) meant for those looking to get into machine learning and who want to understand how the underlying machine learning framework's code works before they are ultra-optimized (which all modern ml libraries are).
+
+```froog``` encapsulates everything from <a href="https://github.com/kevbuh/froog/blob/main/models/linear_regression.py">linear regression</a> to <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">convolutional neural networks </a> in under 1000 lines.
+
+# Installation
+```bash
+pip install froog
+```
+
+More information on downloading ```froog``` in the <a href="https://github.com/kevbuh/froog/blob/main/docs/install.md">installation</a> docs. 
+
+# Features
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/tensor.py">Custom Tensors</a> 
+  - Backpropagation
+  - Automatic Differentiation (autograd)
+      - Forward and backward passes
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/ops.py">ML Operations</a> 
+  - 2D Convolutions (im2col)
+  - Numerical gradient checking
+  - Acceleration methods (Adam)
+  - Avg & Max pooling
+- <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">EfficientNet</a> inference
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/ops_gpu.py">GPU Support</a> 
+- and a bunch <a href="https://github.com/kevbuh/froog/tree/main/froog">more</a> 
+
+# Sneak Peek
+
+Here's how you set up a simple multilayer perceptron for classification on MNIST. Looks pretty similar to pytorch, right?
+
+```python
+from froog.tensor import Tensor
+from froog.nn import Linear
+import froog.optim as optim
+
+class mnistMLP:
+  def __init__(self):
+    self.l1 = Tensor(Linear(784, 128)) # layer 1
+    self.l2 = Tensor(Linear(128, 10))  # layer 2
+
+  def forward(self, x):
+    # forward pass through both layers and softmax for output probabilities
+    return x.dot(self.l1).relu().dot(self.l2).logsoftmax() 
+
+model = mnistMLP() # create model
+optim = optim.SGD([model.l1, model.l2], lr=0.001) # stochastic gradient descent optimizer
+```
+
+# Overview
+
+The most fundamental concept in all of ```froog``` and machine learning frameworks is the <a href="https://github.com/kevbuh/froog/blob/977b09caf32f21904768b08b2772139596604603/froog/tensor.py#L47">Tensor</a>. A <a href="https://en.wikipedia.org/wiki/Tensor_(machine_learning)">tensor</a> is simply a matrix of matrices (more accurately a multi-dimensional array). 
+
+You can create a Tensor in ```froog``` with:
+```python
+import numpy as np
+from froog.tensor import Tensor
+my_tensor = Tensor([1,2,3])
+```
+
+Notice how we had to import numpy. If you want to create a Tensor manually, make sure that it is a Numpy array!
+
+<!-- Learn more about ```froog``` Tensors <a href="https://github.com/kevbuh/froog/blob/main/docs/tensors.md">here</a>. -->
+
+# Tensors
+
+Tensors are the fundamental datatype in froog, and one of the two main classes.
+
+- ```def __init__(self, data)```:
+
+  - Tensor takes in one param, which is the data. Since froog has a numpy backend, the input data into tensors has to be a numpy array.
+
+  - Tensor has a ```self.data``` state that it holds. this contains the data inside of the tensor.
+
+  - In addition, it has ```self.grad```. this is to hold what the gradients of the tensor is. 
+
+  - Lastly, it has ```self._ctx```. theser are the internal vairables used for autograd graph construction. put more simply, this is where the backward gradient computations are saved. 
+
+*Properties*
+
+- ```shape(self)```: this returns the tensor shape
+
+*Methods*
+- ```def zeros(*shape)```: this returns a tensor full of zeros with any shape that you pass in. Defaults to np.float32
+
+- ```def ones(*shape)```: this returns a tensor full of ones with any shape that you pass in. Defaults to np.float32
+
+- ```def randn(*shape):```: this returns a randomly initialized Tensor of *shape
+
+*Gradient calculations*
+
+- ```froog``` computes gradients automatically through a process called automatic differentiation. it has a variable ```_ctx```, which stores the chain of operations. it will take the current operation, lets say a dot product, and go to the dot product definition in ```froog/ops.py```, which contains a backward pass specfically for dot products. all methods, from add to 2x2 maxpools, have this backward pass implemented.
+
+*Functions*
+
+The other base class in froog is the class ```Function```. It keeps track of input tensors and tensors that need to be saved for backward passes
+
+- ```def __init__(self, *tensors)```: takes in an argument of tensors, which are then saved. 
+
+- ```def save_for_backward(self, *x)```: saves Tensors that are necessary to compute for the computation of gradients in the backward pass. 
+
+- ```def apply(self, arg, *x)```: This is what makes everything work. The apply() method takes care of the forward pass, applying the operation to the inputs.
+
+*Register*
+
+```def register(name, fxn)```: this function allows you to add a method to a Tensor. This allows you to chain any operations, e.g. x.dot(w).relu(), where w is a tensor
+
+# Creating a model
+
+Okay cool, so now you know that ```froog```'s main datatype is a Tensor and uses NumPy in the background. How do I actually build a model? 
+
+Here's an example of how to create an MNIST multi-layer perceptron (MLP). We wanted to make it as simple as possible for you to do so so it resembles very basic python concepts like classes. There's really only two methods you need to define: 
+1. ```__init__``` that defines layers of the model (here we use ```Linear```) 
+2. ```forward``` which defines how the input should flow through your model. We use a simple dot product with a ```Linear``` layer with a <a href="https://en.wikipedia.org/wiki/Rectifier_(neural_networks)">```ReLU```</a> activation.
+
+In order to create an instance of the ```mnistMLP``` model, do the same as you would in python: ```model = mnistMLP()``` . 
+
+We support a few different optimizers, <a href="https://github.com/kevbuh/froog/blob/main/froog/optim.py">here</a> which include:
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent">Stochastic Gradient Descent (SGD)</a>
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent#Adam">Adaptive Moment Estimation (Adam)</a>
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent#RMSProp">Root Mean Square Propagation (RMSProp)</a>
+
+```python
+from froog.tensor import Tensor
+import froog.optim as optim
+from froog.nn import Linear
+
+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)
+```
+
+You can also create a convolutional neural net by
+
+```python
+class SimpleConvNet:
+  def __init__(self):
+    conv_size = 5
+    channels = 17
+    self.c1 = Tensor(Linear(channels,1,conv_size,conv_size))     # (num_filters, color_channels, kernel_h, kernel_w)
+    self.l1 = Tensor(Linear((28-conv_size+1)**2*channels, 128))  # (28-conv+1)(28-conv+1) since kernel isn't padded
+    self.l2 = Tensor(Linear(128, 10))                            # MNIST output is 10 classes
+
+  def forward(self, x):
+    x.data = x.data.reshape((-1, 1, 28, 28))                     # get however many number of imgs in batch
+    x = x.conv2d(self.c1).relu()                                 # pass through conv first
+    x = x.reshape(shape=(x.shape[0], -1))
+    return x.dot(self.l1).relu().dot(self.l2).logsoftmax()
+```
+
+So there are two quick examples to get you up and running. You might have noticed some operations like ```reshape``` and were wondering what else you can do with ```froog```. We have many more operations that you can apply on tensors:
+- ```.add()```
+- ```.sub()```
+- ```.mul()```
+- ```.sum()```
+- ```.pow()```
+- ```.dot()```
+- ```.relu()```
+- ```.sigmoid()```
+- ```.reshape()```
+- ```.pad2d()```
+- ```.logsoftmax()```
+- ```.conv2d()```
+- ```.im2col2dconv()```
+- ```.max_pool2d()```
+- ```.avg_pool2d()```
+
+## GPU Support
+
+Have a GPU and need a speedup? You're in good luck because we have GPU support from for our operations defined in <a href="https://github.com/kevbuh/froog/blob/main/froog/ops_gpu.py">```ops_gpu.py```</a>. In order to do this we have a backend built on <a href="https://en.wikipedia.org/wiki/OpenGL">OpenGL</a> that invokes kernel functions that work on the GPU.
+
+Here's how you can send data to the GPU during a forward pass and bring it back to the CPU.
+
+```python
+# ...
+GPU = os.getenv("GPU", None) is not None
+if GPU:
+  out = model.forward(Tensor(img).to_gpu()).cpu()
+```
+
+## EfficientNet in froog!
+
+We have a really cool finished implementation of EfficientNet built entirely in ```froog```!
+
+In order to run EfficientNet inference:
+
+```bash
+VIZ=1 python models/efficientnet.py <https://put_your_image_url_here>
+```
+
+I would recommend checking out the <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">code</a>, it's highly documented and pretty cool. Here's some of the documentation
+```
+Paper           : https://arxiv.org/abs/1905.11946
+PyTorch version : https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/model.py
+
+ConvNets are commonly developed at a fixed resource cost, and then scaled up in order to achieve better accuracy when more resources are made available
+The scaling method was found by performing a grid search to find the relationship between different scaling dimensions of the baseline network under a fixed resource constraint
+"SE" stands for "Squeeze-and-Excitation." Introduced by the "Squeeze-and-Excitation Networks" paper by Jie Hu, Li Shen, and Gang Sun (CVPR 2018).
+
+Environment Variables:
+  VIZ=1 --> plots processed image and output probabilities
+
+How to Run:
+  'VIZ=1 python models/efficientnet.py https://your_image_url'
+
+EfficientNet Hyper-Parameters and Weights:
+url_map = {
+    'efficientnet-b0': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b0-355c32eb.pth',
+    'efficientnet-b1': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b1-f1951068.pth',
+    'efficientnet-b2': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b2-8bb594d6.pth',
+    'efficientnet-b3': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b3-5fb5a3c3.pth',
+    'efficientnet-b4': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b4-6ed6700e.pth',
+    'efficientnet-b5': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b5-b6417697.pth',
+    'efficientnet-b6': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b6-c76e70fd.pth',
+    'efficientnet-b7': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b7-dcc49843.pth',
+}
+
+params_dict = {
+        # Coefficients:   width,depth,res,dropout
+        'efficientnet-b0': (1.0, 1.0, 224, 0.2),
+        'efficientnet-b1': (1.0, 1.1, 240, 0.2),
+        'efficientnet-b2': (1.1, 1.2, 260, 0.3),
+        'efficientnet-b3': (1.2, 1.4, 300, 0.3),
+        'efficientnet-b4': (1.4, 1.8, 380, 0.4),
+        'efficientnet-b5': (1.6, 2.2, 456, 0.4),
+        'efficientnet-b6': (1.8, 2.6, 528, 0.5),
+        'efficientnet-b7': (2.0, 3.1, 600, 0.5),
+        'efficientnet-b8': (2.2, 3.6, 672, 0.5),
+        'efficientnet-l2': (4.3, 5.3, 800, 0.5),
+    }
+
+blocks_args = [
+        'r1_k3_s11_e1_i32_o16_se0.25',
+        'r2_k3_s22_e6_i16_o24_se0.25',
+        'r2_k5_s22_e6_i24_o40_se0.25',
+        'r3_k3_s22_e6_i40_o80_se0.25',
+        'r3_k5_s11_e6_i80_o112_se0.25',
+        'r4_k5_s22_e6_i112_o192_se0.25',
+        'r1_k3_s11_e6_i192_o320_se0.25',
+    ]
+```
+
+## Linear regression 
+
+Doing linear regression in ```froog``` is pretty easy, check out the entire <a href="https://github.com/kevbuh/froog/blob/main/models/linear_regression.py">code</a>.
+
+```bash
+VIZ=1 python3 linear_regression.py
+```
+
+# Contributing
+<!-- THERES LOT OF STUFF TO WORK ON! VISIT THE <a href="https://github.com/kevbuh/froog/blob/main/docs/bounties.md">BOUNTY SHOP</a>  -->
+
+Pull requests will be merged if they:
+* increase simplicity
+* increase functionality
+* increase efficiency
+
+More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributing.md">contributing</a>.
+
+# Documentation
+
+Need more information about how ```froog``` works? Visit the <a href="https://github.com/kevbuh/froog/tree/main/docs">documentation</a>.
+
+# Interested in more?
+
+If you thought ```froog``` was cool, check out the inspirations for this project: pytorch, tinygrad, and https://github.com/karpathy/micrograd/blob/master/micrograd/engine.py

froog-0.3.1/README.md

@@ -0,0 +1,285 @@
+# froog <img src="https://github.com/kevbuh/froog/actions/workflows/test.yml/badge.svg" alt="unit test badge" > <img src="https://static.pepy.tech/badge/froog" alt="num downloads badge">
+<div align="center" >
+  <img src="https://raw.githubusercontent.com/kevbuh/froog/main/assets/froog.png" alt="froog the frog" height="200">
+  <br/>
+  froog: fast real-time optimization of gradients 
+  <br/>
+  a beautifully compact tensor library
+  <br/>
+  <a href="https://github.com/kevbuh/froog">homepage</a> | <a href="https://github.com/kevbuh/froog/tree/main/docs">documentation</a> | <a href="https://pypi.org/project/froog/">pip</a>
+  <br/>
+  <br/>
+</div>
+
+```froog``` is an easy-to-read tensor library (<a href="https://www.pepy.tech/projects/froog">16k pip installs!</a>) meant for those looking to get into machine learning and who want to understand how the underlying machine learning framework's code works before they are ultra-optimized (which all modern ml libraries are).
+
+```froog``` encapsulates everything from <a href="https://github.com/kevbuh/froog/blob/main/models/linear_regression.py">linear regression</a> to <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">convolutional neural networks </a> in under 1000 lines.
+
+# Installation
+```bash
+pip install froog
+```
+
+More information on downloading ```froog``` in the <a href="https://github.com/kevbuh/froog/blob/main/docs/install.md">installation</a> docs. 
+
+# Features
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/tensor.py">Custom Tensors</a> 
+  - Backpropagation
+  - Automatic Differentiation (autograd)
+      - Forward and backward passes
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/ops.py">ML Operations</a> 
+  - 2D Convolutions (im2col)
+  - Numerical gradient checking
+  - Acceleration methods (Adam)
+  - Avg & Max pooling
+- <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">EfficientNet</a> inference
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/ops_gpu.py">GPU Support</a> 
+- and a bunch <a href="https://github.com/kevbuh/froog/tree/main/froog">more</a> 
+
+# Sneak Peek
+
+Here's how you set up a simple multilayer perceptron for classification on MNIST. Looks pretty similar to pytorch, right?
+
+```python
+from froog.tensor import Tensor
+from froog.nn import Linear
+import froog.optim as optim
+
+class mnistMLP:
+  def __init__(self):
+    self.l1 = Tensor(Linear(784, 128)) # layer 1
+    self.l2 = Tensor(Linear(128, 10))  # layer 2
+
+  def forward(self, x):
+    # forward pass through both layers and softmax for output probabilities
+    return x.dot(self.l1).relu().dot(self.l2).logsoftmax() 
+
+model = mnistMLP() # create model
+optim = optim.SGD([model.l1, model.l2], lr=0.001) # stochastic gradient descent optimizer
+```
+
+# Overview
+
+The most fundamental concept in all of ```froog``` and machine learning frameworks is the <a href="https://github.com/kevbuh/froog/blob/977b09caf32f21904768b08b2772139596604603/froog/tensor.py#L47">Tensor</a>. A <a href="https://en.wikipedia.org/wiki/Tensor_(machine_learning)">tensor</a> is simply a matrix of matrices (more accurately a multi-dimensional array). 
+
+You can create a Tensor in ```froog``` with:
+```python
+import numpy as np
+from froog.tensor import Tensor
+my_tensor = Tensor([1,2,3])
+```
+
+Notice how we had to import numpy. If you want to create a Tensor manually, make sure that it is a Numpy array!
+
+<!-- Learn more about ```froog``` Tensors <a href="https://github.com/kevbuh/froog/blob/main/docs/tensors.md">here</a>. -->
+
+# Tensors
+
+Tensors are the fundamental datatype in froog, and one of the two main classes.
+
+- ```def __init__(self, data)```:
+
+  - Tensor takes in one param, which is the data. Since froog has a numpy backend, the input data into tensors has to be a numpy array.
+
+  - Tensor has a ```self.data``` state that it holds. this contains the data inside of the tensor.
+
+  - In addition, it has ```self.grad```. this is to hold what the gradients of the tensor is. 
+
+  - Lastly, it has ```self._ctx```. theser are the internal vairables used for autograd graph construction. put more simply, this is where the backward gradient computations are saved. 
+
+*Properties*
+
+- ```shape(self)```: this returns the tensor shape
+
+*Methods*
+- ```def zeros(*shape)```: this returns a tensor full of zeros with any shape that you pass in. Defaults to np.float32
+
+- ```def ones(*shape)```: this returns a tensor full of ones with any shape that you pass in. Defaults to np.float32
+
+- ```def randn(*shape):```: this returns a randomly initialized Tensor of *shape
+
+*Gradient calculations*
+
+- ```froog``` computes gradients automatically through a process called automatic differentiation. it has a variable ```_ctx```, which stores the chain of operations. it will take the current operation, lets say a dot product, and go to the dot product definition in ```froog/ops.py```, which contains a backward pass specfically for dot products. all methods, from add to 2x2 maxpools, have this backward pass implemented.
+
+*Functions*
+
+The other base class in froog is the class ```Function```. It keeps track of input tensors and tensors that need to be saved for backward passes
+
+- ```def __init__(self, *tensors)```: takes in an argument of tensors, which are then saved. 
+
+- ```def save_for_backward(self, *x)```: saves Tensors that are necessary to compute for the computation of gradients in the backward pass. 
+
+- ```def apply(self, arg, *x)```: This is what makes everything work. The apply() method takes care of the forward pass, applying the operation to the inputs.
+
+*Register*
+
+```def register(name, fxn)```: this function allows you to add a method to a Tensor. This allows you to chain any operations, e.g. x.dot(w).relu(), where w is a tensor
+
+# Creating a model
+
+Okay cool, so now you know that ```froog```'s main datatype is a Tensor and uses NumPy in the background. How do I actually build a model? 
+
+Here's an example of how to create an MNIST multi-layer perceptron (MLP). We wanted to make it as simple as possible for you to do so so it resembles very basic python concepts like classes. There's really only two methods you need to define: 
+1. ```__init__``` that defines layers of the model (here we use ```Linear```) 
+2. ```forward``` which defines how the input should flow through your model. We use a simple dot product with a ```Linear``` layer with a <a href="https://en.wikipedia.org/wiki/Rectifier_(neural_networks)">```ReLU```</a> activation.
+
+In order to create an instance of the ```mnistMLP``` model, do the same as you would in python: ```model = mnistMLP()``` . 
+
+We support a few different optimizers, <a href="https://github.com/kevbuh/froog/blob/main/froog/optim.py">here</a> which include:
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent">Stochastic Gradient Descent (SGD)</a>
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent#Adam">Adaptive Moment Estimation (Adam)</a>
+- <a href="https://en.wikipedia.org/wiki/Stochastic_gradient_descent#RMSProp">Root Mean Square Propagation (RMSProp)</a>
+
+```python
+from froog.tensor import Tensor
+import froog.optim as optim
+from froog.nn import Linear
+
+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)
+```
+
+You can also create a convolutional neural net by
+
+```python
+class SimpleConvNet:
+  def __init__(self):
+    conv_size = 5
+    channels = 17
+    self.c1 = Tensor(Linear(channels,1,conv_size,conv_size))     # (num_filters, color_channels, kernel_h, kernel_w)
+    self.l1 = Tensor(Linear((28-conv_size+1)**2*channels, 128))  # (28-conv+1)(28-conv+1) since kernel isn't padded
+    self.l2 = Tensor(Linear(128, 10))                            # MNIST output is 10 classes
+
+  def forward(self, x):
+    x.data = x.data.reshape((-1, 1, 28, 28))                     # get however many number of imgs in batch
+    x = x.conv2d(self.c1).relu()                                 # pass through conv first
+    x = x.reshape(shape=(x.shape[0], -1))
+    return x.dot(self.l1).relu().dot(self.l2).logsoftmax()
+```
+
+So there are two quick examples to get you up and running. You might have noticed some operations like ```reshape``` and were wondering what else you can do with ```froog```. We have many more operations that you can apply on tensors:
+- ```.add()```
+- ```.sub()```
+- ```.mul()```
+- ```.sum()```
+- ```.pow()```
+- ```.dot()```
+- ```.relu()```
+- ```.sigmoid()```
+- ```.reshape()```
+- ```.pad2d()```
+- ```.logsoftmax()```
+- ```.conv2d()```
+- ```.im2col2dconv()```
+- ```.max_pool2d()```
+- ```.avg_pool2d()```
+
+## GPU Support
+
+Have a GPU and need a speedup? You're in good luck because we have GPU support from for our operations defined in <a href="https://github.com/kevbuh/froog/blob/main/froog/ops_gpu.py">```ops_gpu.py```</a>. In order to do this we have a backend built on <a href="https://en.wikipedia.org/wiki/OpenGL">OpenGL</a> that invokes kernel functions that work on the GPU.
+
+Here's how you can send data to the GPU during a forward pass and bring it back to the CPU.
+
+```python
+# ...
+GPU = os.getenv("GPU", None) is not None
+if GPU:
+  out = model.forward(Tensor(img).to_gpu()).cpu()
+```
+
+## EfficientNet in froog!
+
+We have a really cool finished implementation of EfficientNet built entirely in ```froog```!
+
+In order to run EfficientNet inference:
+
+```bash
+VIZ=1 python models/efficientnet.py <https://put_your_image_url_here>
+```
+
+I would recommend checking out the <a href="https://github.com/kevbuh/froog/blob/main/models/efficientnet.py">code</a>, it's highly documented and pretty cool. Here's some of the documentation
+```
+Paper           : https://arxiv.org/abs/1905.11946
+PyTorch version : https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/model.py
+
+ConvNets are commonly developed at a fixed resource cost, and then scaled up in order to achieve better accuracy when more resources are made available
+The scaling method was found by performing a grid search to find the relationship between different scaling dimensions of the baseline network under a fixed resource constraint
+"SE" stands for "Squeeze-and-Excitation." Introduced by the "Squeeze-and-Excitation Networks" paper by Jie Hu, Li Shen, and Gang Sun (CVPR 2018).
+
+Environment Variables:
+  VIZ=1 --> plots processed image and output probabilities
+
+How to Run:
+  'VIZ=1 python models/efficientnet.py https://your_image_url'
+
+EfficientNet Hyper-Parameters and Weights:
+url_map = {
+    'efficientnet-b0': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b0-355c32eb.pth',
+    'efficientnet-b1': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b1-f1951068.pth',
+    'efficientnet-b2': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b2-8bb594d6.pth',
+    'efficientnet-b3': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b3-5fb5a3c3.pth',
+    'efficientnet-b4': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b4-6ed6700e.pth',
+    'efficientnet-b5': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b5-b6417697.pth',
+    'efficientnet-b6': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b6-c76e70fd.pth',
+    'efficientnet-b7': 'https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b7-dcc49843.pth',
+}
+
+params_dict = {
+        # Coefficients:   width,depth,res,dropout
+        'efficientnet-b0': (1.0, 1.0, 224, 0.2),
+        'efficientnet-b1': (1.0, 1.1, 240, 0.2),
+        'efficientnet-b2': (1.1, 1.2, 260, 0.3),
+        'efficientnet-b3': (1.2, 1.4, 300, 0.3),
+        'efficientnet-b4': (1.4, 1.8, 380, 0.4),
+        'efficientnet-b5': (1.6, 2.2, 456, 0.4),
+        'efficientnet-b6': (1.8, 2.6, 528, 0.5),
+        'efficientnet-b7': (2.0, 3.1, 600, 0.5),
+        'efficientnet-b8': (2.2, 3.6, 672, 0.5),
+        'efficientnet-l2': (4.3, 5.3, 800, 0.5),
+    }
+
+blocks_args = [
+        'r1_k3_s11_e1_i32_o16_se0.25',
+        'r2_k3_s22_e6_i16_o24_se0.25',
+        'r2_k5_s22_e6_i24_o40_se0.25',
+        'r3_k3_s22_e6_i40_o80_se0.25',
+        'r3_k5_s11_e6_i80_o112_se0.25',
+        'r4_k5_s22_e6_i112_o192_se0.25',
+        'r1_k3_s11_e6_i192_o320_se0.25',
+    ]
+```
+
+## Linear regression 
+
+Doing linear regression in ```froog``` is pretty easy, check out the entire <a href="https://github.com/kevbuh/froog/blob/main/models/linear_regression.py">code</a>.
+
+```bash
+VIZ=1 python3 linear_regression.py
+```
+
+# Contributing
+<!-- THERES LOT OF STUFF TO WORK ON! VISIT THE <a href="https://github.com/kevbuh/froog/blob/main/docs/bounties.md">BOUNTY SHOP</a>  -->
+
+Pull requests will be merged if they:
+* increase simplicity
+* increase functionality
+* increase efficiency
+
+More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributing.md">contributing</a>.
+
+# Documentation
+
+Need more information about how ```froog``` works? Visit the <a href="https://github.com/kevbuh/froog/tree/main/docs">documentation</a>.
+
+# Interested in more?
+
+If you thought ```froog``` was cool, check out the inspirations for this project: pytorch, tinygrad, and https://github.com/karpathy/micrograd/blob/master/micrograd/engine.py

{froog-0.2.8 → froog-0.3.1}/froog/ops.py

@@ -52,7 +52,7 @@ class Mul(Function): # x.mul(y)
     return y * grad_output, x * grad_output
 register("mul", Mul)
 
-class Sum(Function):
+class Sum(Function): # x.sum()
   """
   reduce op
   reduces its input tensor to a single value by summing all the elements

{froog-0.2.8 → froog-0.3.1}/froog/tensor.py

@@ -10,10 +10,9 @@
 # inspired by tinygrad
 # inspired by https://github.com/karpathy/micrograd/blob/master/micrograd/engine.py
 
-from functools import partialmethod
-from inspect import signature
-import numpy as np
 import os
+import numpy as np
+from inspect import signature
 
 try:
   import pyopencl as cl

{froog-0.2.8 → froog-0.3.1}/froog/utils.py

@@ -8,29 +8,31 @@
 
 import numpy as np
 from functools import lru_cache
+import pathlib, hashlib, os, tempfile, urllib
 
 def fetch(url):
-  import requests, os, hashlib, tempfile
-  fp = os.path.join(tempfile.gettempdir(), hashlib.md5(url.encode('utf-8')).hexdigest())
-  if os.path.isfile(fp) and os.stat(fp).st_size > 0:
-    print(f"opening cache from {url}...")
-    with open(fp, "rb") as f:
-      dat = f.read()
-  else:
-    print(f"fetching {url}")
-    dat = requests.get(url).content
-    with open(fp+".tmp", "wb") as f:
-      f.write(dat)
-    os.rename(fp+".tmp", fp)
-  return dat
+  if url.startswith(("/", ".")): return pathlib.Path(url)
+  else: fp = pathlib.Path("_cache_dir") / "froog" / "downloads" / (hashlib.md5(url.encode('utf-8')).hexdigest())
+  if not fp.is_file():
+    with urllib.request.urlopen(url, timeout=10) as r:
+      assert r.status == 200
+      total_length = int(r.headers.get('content-length', 0))
+      (path := fp.parent).mkdir(parents=True, exist_ok=True)
+      with tempfile.NamedTemporaryFile(dir=path, delete=False) as f:
+        while chunk := r.read(16384): f.write(chunk)
+        f.close()
+        if (file_size:=os.stat(f.name).st_size) < total_length: raise RuntimeError(f"fetch size incomplete, {file_size} < {total_length}")
+        pathlib.Path(f.name).rename(fp)
+  return fp
 
 def fetch_mnist():
   import gzip
-  parse = lambda dat: np.frombuffer(gzip.decompress(dat), dtype=np.uint8).copy()
-  X_train = parse(fetch("http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz"))[0x10:].reshape((-1, 28, 28))
-  Y_train = parse(fetch("http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz"))[8:]
-  X_test = parse(fetch("http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz"))[0x10:].reshape((-1, 28, 28))
-  Y_test = parse(fetch("http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz"))[8:]
+  parse = lambda file: np.frombuffer(gzip.open(file).read(), dtype=np.uint8).copy()
+  BASE_URL = "https://storage.googleapis.com/cvdf-datasets/mnist/"
+  X_train = parse(fetch(f"{BASE_URL}train-images-idx3-ubyte.gz"))[0x10:].reshape((-1, 28*28)).astype(np.float32)
+  Y_train = parse(fetch(f"{BASE_URL}train-labels-idx1-ubyte.gz"))[8:].astype(np.int8)
+  X_test = parse(fetch(f"{BASE_URL}t10k-images-idx3-ubyte.gz"))[0x10:].reshape((-1, 28*28)).astype(np.float32)
+  Y_test = parse(fetch(f"{BASE_URL}t10k-labels-idx1-ubyte.gz"))[8:].astype(np.int8)
   return X_train, Y_train, X_test, Y_test
 
 def mask_like(like, mask_inx, mask_value=1.0):