clax

handy classifiers in jax

Installation

pip install .

For Apple silicon users with Sonoma 14.4+ wanting to try GPU acceleration (may need ENABLE_PJRT_COMPATIBILITY=1)

pip install ".[metal]"

or try equivalently from pypi

Examples included in examples directory

Name		Name	Last commit message	Last commit date
Latest commit History 38 Commits
.github/workflows		.github/workflows
clax		clax
examples		examples
tests		tests
.flake8		.flake8
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
LICENSE		LICENSE
README.md		README.md
pyproject.toml		pyproject.toml
setup.cfg		setup.cfg