Metadata-Version: 2.4
Name: torchfoo
Version: 0.0.3
Summary: A collection of general utilities for PyTorch
Author-email: Vinam Arora <vinam@seas.upenn.edu>
License-Expression: MIT
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: torch>=2.4.0
Provides-Extra: dev
Requires-Dist: numpy; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: black==24.2.0; extra == "dev"
Provides-Extra: docs
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: pydata-sphinx-theme; extra == "docs"

# torchfoo

[![PyPI version](https://badge.fury.io/py/torchfoo.svg)](https://badge.fury.io/py/torchfoo)
[![Documentation](https://img.shields.io/badge/docs-readthedocs-blue)](https://torchfoo.readthedocs.io)

A collection of utilities for PyTorch

## Installation

Requires Python >= 3.10 and PyTorch >= 2.4.0.

```bash
pip install torchfoo
```

### Development

```bash
git clone https://github.com/vinamarora8/torchfoo.git
cd torchfoo
python -m venv venv
source venv/bin/activate
pip install -e ".[dev]"
```
