Metadata-Version: 2.4
Name: frogbox
Version: 0.6.0
Summary: An opinionated machine learning framework
Author-email: "Simon J. Larsen" <simonhffh@gmail.com>
License-Expression: MIT
Project-URL: Homepage, https://github.com/SimonLarsen/frogbox
Project-URL: Issues, https://github.com/SimonLarsen/frogbox/issues
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch>=2.5.1
Requires-Dist: torchvision>=0.20.1
Requires-Dist: accelerate>=1.12.0
Requires-Dist: torchmetrics>=1.8.2
Requires-Dist: pydantic>=2.12.4
Requires-Dist: tqdm>=4.67.1
Requires-Dist: jinja2>=3.1.6
Requires-Dist: click>=8.3.1
Requires-Dist: PyYAML>=6.0.3
Dynamic: license-file

<p align="center">
    <img src="https://simonlarsen.github.io/frogbox/logo.png" width="224">
</p>
<h1 align="center">
    frogbox
</h1>

Frogbox is an opinionated PyTorch machine learning framework built for rapid prototyping and research.

## Features

* Experiments are defined using JSON files and support [jinja2](https://jinja.palletsprojects.com) templates.
* Flexible event system inspired by [Ignite](https://pytorch.org/ignite).
* Automatic experiment tracking. Currently, [Weights & Biases](https://wandb.ai) and [MLFlow](https://mlflow.org) is supported with other platforms planned.
* CLI tool for easy project management.
* Integrates [Accelerate](https://huggingface.co/docs/accelerate/index) to support automatic mixed precision (AMP) and distributed training.

## Installation

```sh
pip install frogbox
```

## Getting started

See [documentation](https://frogbox.readthedocs.io).
