Metadata-Version: 2.4
Name: frogbox
Version: 0.5.3
Summary: An opinionated machine learning framework
Author-email: "Simon J. Larsen" <simonhffh@gmail.com>
License-Expression: MIT
Project-URL: Homepage, https://hisimon.dk/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.2.0
Requires-Dist: torchmetrics>=1.6.0
Requires-Dist: pydantic>=2.10.3
Requires-Dist: tqdm>=4.67.1
Requires-Dist: wandb>=0.18.7
Requires-Dist: jinja2>=3.1.4
Requires-Dist: click>=8.1.7
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 only [Weights & Biases](https://wandb.ai/) is supported with other platforms planned.
* CLI tool for easy project management. Just type `frogbox project new -t supervised` to get started.
* Integrates [Accelerate](https://huggingface.co/docs/accelerate/index) to support automatic mixed precision (AMP) and distributed training.

## Installation

```sh
pip install git+https://SimonLarsen@github.com/SimonLarsen/frogbox.git@v0.5.3
```

## Getting started

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