Metadata-Version: 2.4
Name: ionix
Version: 0.0
Summary: Add your description here
Author-email: Benedict Armstrong <benedict.armstrong@inf.ethz.ch>
Maintainer: Benedict Armstrong
Requires-Python: >=3.12
Requires-Dist: flax
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: matplotlib>=3.10.7
Requires-Dist: optax
Requires-Dist: tqdm
Provides-Extra: dev
Requires-Dist: pre-commit>=4.3.0; extra == 'dev'
Requires-Dist: pytest>=8.3.3; extra == 'dev'
Requires-Dist: ruff>=0.12.11; extra == 'dev'
Description-Content-Type: text/markdown

# Ionix

Ionix is a library for training JAX models. It should be familiar anyone who has used Pytorch Lightning.

Ionix defines a set of interfaces for training and provides some reference implementations for common tasks while remaining as minimal and expandable as possible

## Installation

```bash
uv add ionix
```

or using pip:

```bash
pip install ionix
```
