froog 0.5.0__tar.gz → 0.5.2__tar.gz

{froog-0.5.0 → froog-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: froog
-Version: 0.5.0
+Version: 0.5.2
 Summary: tensor library with opencl and metal support
 Author: Kevin Buhler
 License: MIT
@@ -13,23 +13,21 @@ Description-Content-Type: text/markdown
 <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
+  froog: a gpu accelerated tensor library
   <br/>
   <a href="https://github.com/kevbuh/froog">homepage</a> | <a href="https://github.com/kevbuh/froog/tree/main/DOCS.md">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">25k pip installs!</a>) with OpenCL support for GPU acceleration. Inspired by pytorch, tinygrad, and micrograd.
+```froog``` is an easy-to-read tensor library (<a href="https://www.pepy.tech/projects/froog">27k pip installs!</a>) with support for GPU acceleration with [OpenCL](https://www.khronos.org/opencl/) and [Apple Metal](https://developer.apple.com/metal/). Inspired by [tinygrad](https://github.com/tinygrad/tinygrad), and [micrograd](https://github.com/karpathy/micrograd).
 
-# Installation
+## Installation
 ```bash
 pip install froog
 ```
 
-# Features
+## Features
 - <a href="https://github.com/kevbuh/froog/blob/main/froog/tensor.py">Custom Tensors</a> 
   - Backpropagation
   - Automatic Differentiation (autograd)
@@ -40,11 +38,9 @@ pip install froog
   - 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/gpu/ops_gpu.py">GPU Support</a> 
-- <a href="https://github.com/kevbuh/froog/blob/main/docs/env.md">Configuration via Environment Variables</a>
-- and a bunch <a href="https://github.com/kevbuh/froog/tree/main/froog">more</a> 
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/gpu">GPU Support</a> 
 
-# Example
+## Quick Example
 
 Here's how you set up a simple multilayer perceptron for classification on MNIST. Looks pretty similar to pytorch, right?
 
@@ -66,7 +62,7 @@ model = mnistMLP() # create model
 optim = optim.SGD([model.l1, model.l2], lr=0.001) # stochastic gradient descent optimizer
 ```
 
-# GPU Support
+## GPU Support
 
 Device management is handled transparently and will automatically select one of ```[METAL, OPENCL, CPU]```. To use the GPU:
 
@@ -102,11 +98,12 @@ from froog import set_device
 set_device("METAL")  # or "OPENCL"
 ```
 
-# EfficientNet in froog!
+## EfficientNet in froog!
+
+<img src="https://github.com/kevbuh/froog/blob/main/assets/efficientnet_pug.png" alt="pug" height="200">
 
-<img src="assets/efficientnet_pug.png" alt="pug" height="300">
+We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Running inference on this pug...
 
-We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Run inference with:
 
 ```bash
 python3 models/efficientnet.py <https://optional_image_url>
@@ -116,25 +113,25 @@ inference 4.34 s
 
 imagenet class: 254
 prediction    : pug, pug-dog
-probability   : 0.8902361
+probability   : 0.9402361
 ******************************
 ```
 
 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.
 
-# Contributing
+<!-- ## 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:
+<!-- 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>. Make sure to run ```python -m pytest``` before creating a PR.
+More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributing.md">contributing</a>. Make sure to run ```python -m pytest``` before creating a PR. -->
 
-# API
+## API
 
-## Basic Math Operations
+MATH
 - ```.add(y)``` - Addition with y
 - ```.sub(y)``` - Subtraction with y
 - ```.mul(y)``` - Multiplication with y
@@ -143,12 +140,10 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.sum()``` - Sum all elements
 - ```.mean()``` - Mean of all elements
 - ```.sqrt()``` - Square root
-
-## Linear Algebra Operations
 - ```.dot(y)``` - Matrix multiplication with y
 - ```.matmul(y)``` - Alias for dot
 
-## Neural Network Operations
+MACHINE LEARNING
 - ```.relu()``` - Rectified Linear Unit activation
 - ```.sigmoid()``` - Sigmoid activation
 - ```.dropout(p=0.5, training=True)``` - Dropout regularization
@@ -156,22 +151,17 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.swish()``` - Swish activation function (x * sigmoid(x))
 - ```.conv2d(w, stride=1, groups=1)``` - 2D convolution
 - ```.im2col2dconv(w)``` - Image to column for convolution
-
-## Pooling Operations
 - ```.max_pool2d(kernel_size=(2,2))``` - 2D max pooling
 - ```.avg_pool2d(kernel_size=(2,2))``` - 2D average pooling
 
-## Tensor Manipulation
-- ```.reshape(*shape)``` - Change tensor shape
-- ```.view(*shape)``` - Alternative to reshape
-- ```.pad2d(padding=None)``` - Pad 2D tensors
-- ```.flatten()``` - Returns a flattened 1D copy of the tensor
-- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
-- ```.squeeze(dim=None)``` - Remove dimensions of size 1
-- ```.detach()``` - Returns a tensor detached from computation graph
-- ```.assign(x)``` - Assign values from tensor x to this tensor
+TENSOR
+- ```Tensor.zeros(*shape)``` - Create tensor of zeros
+- ```Tensor.ones(*shape)``` - Create tensor of ones
+- ```Tensor.randn(*shape)``` - Create tensor with random normal values
+- ```Tensor.eye(dim)``` - Create identity matrix
+- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values
 
-## Tensor Properties
+TENSOR PROPERTIES
 - ```.shape``` - The shape of the tensor as a tuple
 - ```.size``` - Total number of elements in the tensor
 - ```.ndim``` - Number of dimensions (rank) of the tensor
@@ -180,23 +170,22 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.is_gpu``` - Whether tensor is on GPU
 - ```.grad``` - Gradient of tensor with respect to some scalar value
 - ```.data``` - Underlying NumPy array (or GPU buffer)
+- ```.to_float()``` - Converts tensor to float32 data type
+- ```.to_int()``` - Converts tensor to int32 data type
+- ```.to_bool()``` - Converts tensor to boolean data type
+- ```.reshape(*shape)``` - Change tensor shape
+- ```.view(*shape)``` - Alternative to reshape
+- ```.pad2d(padding=None)``` - Pad 2D tensors
+- ```.flatten()``` - Returns a flattened 1D copy of the tensor
+- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
+- ```.squeeze(dim=None)``` - Remove dimensions of size 1
+- ```.detach()``` - Returns a tensor detached from computation graph
+- ```.assign(x)``` - Assign values from tensor x to this tensor
 
-## Device Management
+GPU
 - ```.to_cpu()``` - Moves tensor to CPU
 - ```.to_gpu()``` - Moves tensor to GPU 
 - ```.gpu_()``` - In-place GPU conversion (modifies tensor)
 
-## Data Type Conversion
-- ```.to_float()``` - Converts tensor to float32 data type
-- ```.to_int()``` - Converts tensor to int32 data type
-- ```.to_bool()``` - Converts tensor to boolean data type
-
-## Autograd Operations
+AUTOGRAD
 - ```.backward(allow_fill=True)``` - Performs backpropagation
-
-## Tensor Creation Methods
-- ```Tensor.zeros(*shape)``` - Create tensor of zeros
-- ```Tensor.ones(*shape)``` - Create tensor of ones
-- ```Tensor.randn(*shape)``` - Create tensor with random normal values
-- ```Tensor.eye(dim)``` - Create identity matrix
-- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values

{froog-0.5.0 → froog-0.5.2}/README.md

@@ -2,23 +2,21 @@
 <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
+  froog: a gpu accelerated tensor library
   <br/>
   <a href="https://github.com/kevbuh/froog">homepage</a> | <a href="https://github.com/kevbuh/froog/tree/main/DOCS.md">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">25k pip installs!</a>) with OpenCL support for GPU acceleration. Inspired by pytorch, tinygrad, and micrograd.
+```froog``` is an easy-to-read tensor library (<a href="https://www.pepy.tech/projects/froog">27k pip installs!</a>) with support for GPU acceleration with [OpenCL](https://www.khronos.org/opencl/) and [Apple Metal](https://developer.apple.com/metal/). Inspired by [tinygrad](https://github.com/tinygrad/tinygrad), and [micrograd](https://github.com/karpathy/micrograd).
 
-# Installation
+## Installation
 ```bash
 pip install froog
 ```
 
-# Features
+## Features
 - <a href="https://github.com/kevbuh/froog/blob/main/froog/tensor.py">Custom Tensors</a> 
   - Backpropagation
   - Automatic Differentiation (autograd)
@@ -29,11 +27,9 @@ pip install froog
   - 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/gpu/ops_gpu.py">GPU Support</a> 
-- <a href="https://github.com/kevbuh/froog/blob/main/docs/env.md">Configuration via Environment Variables</a>
-- and a bunch <a href="https://github.com/kevbuh/froog/tree/main/froog">more</a> 
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/gpu">GPU Support</a> 
 
-# Example
+## Quick Example
 
 Here's how you set up a simple multilayer perceptron for classification on MNIST. Looks pretty similar to pytorch, right?
 
@@ -55,7 +51,7 @@ model = mnistMLP() # create model
 optim = optim.SGD([model.l1, model.l2], lr=0.001) # stochastic gradient descent optimizer
 ```
 
-# GPU Support
+## GPU Support
 
 Device management is handled transparently and will automatically select one of ```[METAL, OPENCL, CPU]```. To use the GPU:
 
@@ -91,11 +87,12 @@ from froog import set_device
 set_device("METAL")  # or "OPENCL"
 ```
 
-# EfficientNet in froog!
+## EfficientNet in froog!
+
+<img src="https://github.com/kevbuh/froog/blob/main/assets/efficientnet_pug.png" alt="pug" height="200">
 
-<img src="assets/efficientnet_pug.png" alt="pug" height="300">
+We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Running inference on this pug...
 
-We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Run inference with:
 
 ```bash
 python3 models/efficientnet.py <https://optional_image_url>
@@ -105,25 +102,25 @@ inference 4.34 s
 
 imagenet class: 254
 prediction    : pug, pug-dog
-probability   : 0.8902361
+probability   : 0.9402361
 ******************************
 ```
 
 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.
 
-# Contributing
+<!-- ## 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:
+<!-- 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>. Make sure to run ```python -m pytest``` before creating a PR.
+More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributing.md">contributing</a>. Make sure to run ```python -m pytest``` before creating a PR. -->
 
-# API
+## API
 
-## Basic Math Operations
+MATH
 - ```.add(y)``` - Addition with y
 - ```.sub(y)``` - Subtraction with y
 - ```.mul(y)``` - Multiplication with y
@@ -132,12 +129,10 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.sum()``` - Sum all elements
 - ```.mean()``` - Mean of all elements
 - ```.sqrt()``` - Square root
-
-## Linear Algebra Operations
 - ```.dot(y)``` - Matrix multiplication with y
 - ```.matmul(y)``` - Alias for dot
 
-## Neural Network Operations
+MACHINE LEARNING
 - ```.relu()``` - Rectified Linear Unit activation
 - ```.sigmoid()``` - Sigmoid activation
 - ```.dropout(p=0.5, training=True)``` - Dropout regularization
@@ -145,22 +140,17 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.swish()``` - Swish activation function (x * sigmoid(x))
 - ```.conv2d(w, stride=1, groups=1)``` - 2D convolution
 - ```.im2col2dconv(w)``` - Image to column for convolution
-
-## Pooling Operations
 - ```.max_pool2d(kernel_size=(2,2))``` - 2D max pooling
 - ```.avg_pool2d(kernel_size=(2,2))``` - 2D average pooling
 
-## Tensor Manipulation
-- ```.reshape(*shape)``` - Change tensor shape
-- ```.view(*shape)``` - Alternative to reshape
-- ```.pad2d(padding=None)``` - Pad 2D tensors
-- ```.flatten()``` - Returns a flattened 1D copy of the tensor
-- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
-- ```.squeeze(dim=None)``` - Remove dimensions of size 1
-- ```.detach()``` - Returns a tensor detached from computation graph
-- ```.assign(x)``` - Assign values from tensor x to this tensor
+TENSOR
+- ```Tensor.zeros(*shape)``` - Create tensor of zeros
+- ```Tensor.ones(*shape)``` - Create tensor of ones
+- ```Tensor.randn(*shape)``` - Create tensor with random normal values
+- ```Tensor.eye(dim)``` - Create identity matrix
+- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values
 
-## Tensor Properties
+TENSOR PROPERTIES
 - ```.shape``` - The shape of the tensor as a tuple
 - ```.size``` - Total number of elements in the tensor
 - ```.ndim``` - Number of dimensions (rank) of the tensor
@@ -169,23 +159,22 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.is_gpu``` - Whether tensor is on GPU
 - ```.grad``` - Gradient of tensor with respect to some scalar value
 - ```.data``` - Underlying NumPy array (or GPU buffer)
+- ```.to_float()``` - Converts tensor to float32 data type
+- ```.to_int()``` - Converts tensor to int32 data type
+- ```.to_bool()``` - Converts tensor to boolean data type
+- ```.reshape(*shape)``` - Change tensor shape
+- ```.view(*shape)``` - Alternative to reshape
+- ```.pad2d(padding=None)``` - Pad 2D tensors
+- ```.flatten()``` - Returns a flattened 1D copy of the tensor
+- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
+- ```.squeeze(dim=None)``` - Remove dimensions of size 1
+- ```.detach()``` - Returns a tensor detached from computation graph
+- ```.assign(x)``` - Assign values from tensor x to this tensor
 
-## Device Management
+GPU
 - ```.to_cpu()``` - Moves tensor to CPU
 - ```.to_gpu()``` - Moves tensor to GPU 
 - ```.gpu_()``` - In-place GPU conversion (modifies tensor)
 
-## Data Type Conversion
-- ```.to_float()``` - Converts tensor to float32 data type
-- ```.to_int()``` - Converts tensor to int32 data type
-- ```.to_bool()``` - Converts tensor to boolean data type
-
-## Autograd Operations
+AUTOGRAD
 - ```.backward(allow_fill=True)``` - Performs backpropagation
-
-## Tensor Creation Methods
-- ```Tensor.zeros(*shape)``` - Create tensor of zeros
-- ```Tensor.ones(*shape)``` - Create tensor of ones
-- ```Tensor.randn(*shape)``` - Create tensor with random normal values
-- ```Tensor.eye(dim)``` - Create identity matrix
-- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values

{froog-0.5.0 → froog-0.5.2}/froog.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: froog
-Version: 0.5.0
+Version: 0.5.2
 Summary: tensor library with opencl and metal support
 Author: Kevin Buhler
 License: MIT
@@ -13,23 +13,21 @@ Description-Content-Type: text/markdown
 <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
+  froog: a gpu accelerated tensor library
   <br/>
   <a href="https://github.com/kevbuh/froog">homepage</a> | <a href="https://github.com/kevbuh/froog/tree/main/DOCS.md">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">25k pip installs!</a>) with OpenCL support for GPU acceleration. Inspired by pytorch, tinygrad, and micrograd.
+```froog``` is an easy-to-read tensor library (<a href="https://www.pepy.tech/projects/froog">27k pip installs!</a>) with support for GPU acceleration with [OpenCL](https://www.khronos.org/opencl/) and [Apple Metal](https://developer.apple.com/metal/). Inspired by [tinygrad](https://github.com/tinygrad/tinygrad), and [micrograd](https://github.com/karpathy/micrograd).
 
-# Installation
+## Installation
 ```bash
 pip install froog
 ```
 
-# Features
+## Features
 - <a href="https://github.com/kevbuh/froog/blob/main/froog/tensor.py">Custom Tensors</a> 
   - Backpropagation
   - Automatic Differentiation (autograd)
@@ -40,11 +38,9 @@ pip install froog
   - 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/gpu/ops_gpu.py">GPU Support</a> 
-- <a href="https://github.com/kevbuh/froog/blob/main/docs/env.md">Configuration via Environment Variables</a>
-- and a bunch <a href="https://github.com/kevbuh/froog/tree/main/froog">more</a> 
+- <a href="https://github.com/kevbuh/froog/blob/main/froog/gpu">GPU Support</a> 
 
-# Example
+## Quick Example
 
 Here's how you set up a simple multilayer perceptron for classification on MNIST. Looks pretty similar to pytorch, right?
 
@@ -66,7 +62,7 @@ model = mnistMLP() # create model
 optim = optim.SGD([model.l1, model.l2], lr=0.001) # stochastic gradient descent optimizer
 ```
 
-# GPU Support
+## GPU Support
 
 Device management is handled transparently and will automatically select one of ```[METAL, OPENCL, CPU]```. To use the GPU:
 
@@ -102,11 +98,12 @@ from froog import set_device
 set_device("METAL")  # or "OPENCL"
 ```
 
-# EfficientNet in froog!
+## EfficientNet in froog!
+
+<img src="https://github.com/kevbuh/froog/blob/main/assets/efficientnet_pug.png" alt="pug" height="200">
 
-<img src="assets/efficientnet_pug.png" alt="pug" height="300">
+We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Running inference on this pug...
 
-We have an implementation of [EfficientNet v2](https://arxiv.org/abs/2104.00298) built entirely in ```froog``` using the official PyTorch weights! Run inference with:
 
 ```bash
 python3 models/efficientnet.py <https://optional_image_url>
@@ -116,25 +113,25 @@ inference 4.34 s
 
 imagenet class: 254
 prediction    : pug, pug-dog
-probability   : 0.8902361
+probability   : 0.9402361
 ******************************
 ```
 
 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.
 
-# Contributing
+<!-- ## 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:
+<!-- 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>. Make sure to run ```python -m pytest``` before creating a PR.
+More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributing.md">contributing</a>. Make sure to run ```python -m pytest``` before creating a PR. -->
 
-# API
+## API
 
-## Basic Math Operations
+MATH
 - ```.add(y)``` - Addition with y
 - ```.sub(y)``` - Subtraction with y
 - ```.mul(y)``` - Multiplication with y
@@ -143,12 +140,10 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.sum()``` - Sum all elements
 - ```.mean()``` - Mean of all elements
 - ```.sqrt()``` - Square root
-
-## Linear Algebra Operations
 - ```.dot(y)``` - Matrix multiplication with y
 - ```.matmul(y)``` - Alias for dot
 
-## Neural Network Operations
+MACHINE LEARNING
 - ```.relu()``` - Rectified Linear Unit activation
 - ```.sigmoid()``` - Sigmoid activation
 - ```.dropout(p=0.5, training=True)``` - Dropout regularization
@@ -156,22 +151,17 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.swish()``` - Swish activation function (x * sigmoid(x))
 - ```.conv2d(w, stride=1, groups=1)``` - 2D convolution
 - ```.im2col2dconv(w)``` - Image to column for convolution
-
-## Pooling Operations
 - ```.max_pool2d(kernel_size=(2,2))``` - 2D max pooling
 - ```.avg_pool2d(kernel_size=(2,2))``` - 2D average pooling
 
-## Tensor Manipulation
-- ```.reshape(*shape)``` - Change tensor shape
-- ```.view(*shape)``` - Alternative to reshape
-- ```.pad2d(padding=None)``` - Pad 2D tensors
-- ```.flatten()``` - Returns a flattened 1D copy of the tensor
-- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
-- ```.squeeze(dim=None)``` - Remove dimensions of size 1
-- ```.detach()``` - Returns a tensor detached from computation graph
-- ```.assign(x)``` - Assign values from tensor x to this tensor
+TENSOR
+- ```Tensor.zeros(*shape)``` - Create tensor of zeros
+- ```Tensor.ones(*shape)``` - Create tensor of ones
+- ```Tensor.randn(*shape)``` - Create tensor with random normal values
+- ```Tensor.eye(dim)``` - Create identity matrix
+- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values
 
-## Tensor Properties
+TENSOR PROPERTIES
 - ```.shape``` - The shape of the tensor as a tuple
 - ```.size``` - Total number of elements in the tensor
 - ```.ndim``` - Number of dimensions (rank) of the tensor
@@ -180,23 +170,22 @@ More info on <a href="https://github.com/kevbuh/froog/blob/main/docs/contributin
 - ```.is_gpu``` - Whether tensor is on GPU
 - ```.grad``` - Gradient of tensor with respect to some scalar value
 - ```.data``` - Underlying NumPy array (or GPU buffer)
+- ```.to_float()``` - Converts tensor to float32 data type
+- ```.to_int()``` - Converts tensor to int32 data type
+- ```.to_bool()``` - Converts tensor to boolean data type
+- ```.reshape(*shape)``` - Change tensor shape
+- ```.view(*shape)``` - Alternative to reshape
+- ```.pad2d(padding=None)``` - Pad 2D tensors
+- ```.flatten()``` - Returns a flattened 1D copy of the tensor
+- ```.unsqueeze(dim)``` - Add dimension of size 1 at specified position
+- ```.squeeze(dim=None)``` - Remove dimensions of size 1
+- ```.detach()``` - Returns a tensor detached from computation graph
+- ```.assign(x)``` - Assign values from tensor x to this tensor
 
-## Device Management
+GPU
 - ```.to_cpu()``` - Moves tensor to CPU
 - ```.to_gpu()``` - Moves tensor to GPU 
 - ```.gpu_()``` - In-place GPU conversion (modifies tensor)
 
-## Data Type Conversion
-- ```.to_float()``` - Converts tensor to float32 data type
-- ```.to_int()``` - Converts tensor to int32 data type
-- ```.to_bool()``` - Converts tensor to boolean data type
-
-## Autograd Operations
+AUTOGRAD
 - ```.backward(allow_fill=True)``` - Performs backpropagation
-
-## Tensor Creation Methods
-- ```Tensor.zeros(*shape)``` - Create tensor of zeros
-- ```Tensor.ones(*shape)``` - Create tensor of ones
-- ```Tensor.randn(*shape)``` - Create tensor with random normal values
-- ```Tensor.eye(dim)``` - Create identity matrix
-- ```Tensor.arange(start, stop=None, step=1)``` - Create tensor with evenly spaced values

{froog-0.5.0 → froog-0.5.2}/setup.py

@@ -9,7 +9,7 @@ with open(os.path.join(directory, 'README.md'), encoding='utf-8') as f:
   long_description = f.read()
 
 setup(name='froog',
-      version='0.5.0',
+      version='0.5.2',
       description='tensor library with opencl and metal support',
       author='Kevin Buhler',
       license='MIT',