# torch-batteries

<p align="center">
   <a href="https://pypi.org/project/torch-batteries/"><img src="https://img.shields.io/pypi/v/torch-batteries" alt="PyPI version"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-MIT-blue.svg" alt="License: MIT"></a>
</p>

<p align="center">
   <img src="https://raw.githubusercontent.com/michalszc/torch-batteries/refs/heads/master/assets/logo.png" alt="torch-batteries" style="max-height:512px; height:auto;" />
   <br/>
   <em>Image generated by AI</em>
</p>

A lightweight Python package that supplies batteries-included abstractions for:
- Data loading pipelines
- Model training loops
- Evaluation workflows
- Metrics computation
- Seamless Weights & Biases tracking

Designed to reduce boilerplate and standardize experiment code.

## Installation

```bash
pip install torch-batteries
```
