Metadata-Version: 2.4
Name: xax
Version: 0.2.12
Summary: A library for fast Jax experimentation
Home-page: https://github.com/kscalelabs/xax
Author: Benjamin Bolte
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: attrs
Requires-Dist: chex
Requires-Dist: dpshdl
Requires-Dist: equinox
Requires-Dist: importlib-resources
Requires-Dist: jax
Requires-Dist: jaxtyping
Requires-Dist: optax
Requires-Dist: orbax-checkpoint
Requires-Dist: pillow
Requires-Dist: omegaconf
Requires-Dist: gitpython
Requires-Dist: tensorboard
Requires-Dist: psutil
Requires-Dist: requests
Provides-Extra: dev
Requires-Dist: black; extra == "dev"
Requires-Dist: darglint; extra == "dev"
Requires-Dist: mypy; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: types-pillow; extra == "dev"
Requires-Dist: types-psutil; extra == "dev"
Requires-Dist: types-requests; extra == "dev"
Provides-Extra: exportable
Requires-Dist: flax; extra == "exportable"
Requires-Dist: orbax-export; extra == "exportable"
Requires-Dist: tensorflow; extra == "exportable"
Provides-Extra: all
Requires-Dist: black; extra == "all"
Requires-Dist: darglint; extra == "all"
Requires-Dist: mypy; extra == "all"
Requires-Dist: ruff; extra == "all"
Requires-Dist: pytest; extra == "all"
Requires-Dist: types-pillow; extra == "all"
Requires-Dist: types-psutil; extra == "all"
Requires-Dist: types-requests; extra == "all"
Requires-Dist: flax; extra == "all"
Requires-Dist: orbax-export; extra == "all"
Requires-Dist: tensorflow; extra == "all"
Dynamic: author
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license-file
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

# xax

JAX library for fast experimentation.

## Installation

```bash
pip install xax
```
