Metadata-Version: 2.4
Name: volara-ml
Version: 1.1.0
Summary: Plugin for Volara including ML-related tasks. This includes the `Predict` task and the `models` module.
Requires-Python: >=3.11
Description-Content-Type: text/markdown
Requires-Dist: volara>=1.0.2
Requires-Dist: daisy>=1.2.2
Requires-Dist: gunpowder>=1.4.0
Requires-Dist: funlib.geometry>=0.3
Requires-Dist: funlib.persistence>=0.6
Requires-Dist: pydantic>=2.6.3
Provides-Extra: torch
Requires-Dist: torch>=2.4; extra == "torch"
Provides-Extra: jax
Requires-Dist: jax>=0.4.30; extra == "jax"
Requires-Dist: flax>=0.8; extra == "jax"
Requires-Dist: orbax-checkpoint>=0.5; extra == "jax"
Provides-Extra: dev
Requires-Dist: pytest>=8.0; extra == "dev"
Requires-Dist: pytest-cov>=5.0; extra == "dev"
Requires-Dist: ruff>=0.6; extra == "dev"
Requires-Dist: torch>=2.4; extra == "dev"
Requires-Dist: ty>=0.0.24; extra == "dev"
Provides-Extra: docs
Requires-Dist: autodoc_pydantic>=2.0; extra == "docs"
Requires-Dist: sphinx-autodoc-typehints>=2.0; extra == "docs"
Requires-Dist: sphinx-material>=0.0.36; extra == "docs"
Requires-Dist: sphinx-click>=6.0; extra == "docs"
Requires-Dist: sphinx-rtd-theme>=2.0; extra == "docs"
Requires-Dist: myst-parser>=4.0; extra == "docs"
Requires-Dist: jupyter-sphinx>=0.5; extra == "docs"
Requires-Dist: matplotlib>=3.0; extra == "docs"
Requires-Dist: pooch>=1.8; extra == "docs"
Requires-Dist: wget>=3.2; extra == "docs"
Requires-Dist: funlib-show-neuroglancer>=0.2; extra == "docs"
Requires-Dist: nbsphinx>=0.9.7; extra == "docs"
Requires-Dist: jupytext>=1.16.7; extra == "docs"

[![tests](https://github.com/e11bio/volara-ml/actions/workflows/tests.yaml/badge.svg)](https://github.com/e11bio/volara-ml/actions/workflows/tests.yaml)
[![ruff](https://github.com/e11bio/volara-ml/actions/workflows/ruff.yaml/badge.svg)](https://github.com/e11bio/volara-ml/actions/workflows/ruff.yaml)
[![ty](https://github.com/e11bio/volara-ml/actions/workflows/ty.yaml/badge.svg)](https://github.com/e11bio/volara-ml/actions/workflows/ty.yaml)
<!-- [![codecov](https://codecov.io/gh/e11bio/volara-ml/branch/main/graph/badge.svg?token=YOUR_TOKEN)](https://codecov.io/gh/e11bio/volara-ml) -->

# volara-ml
A plugin for volara that includes ML models and a `Predict` task

## Installation

```bash
pip install volara-ml         # base package
pip install volara-ml[torch]  # with PyTorch support
pip install volara-ml[jax]    # with Jax support
```

# Available blockwise operations:
- `Predict`: Model Prediction (requires `torch` or `jax` extra)
