Prebuilt jax classifiers
Project description
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
Examples included in examples
directory
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
clax-0.0.6.tar.gz
(5.9 kB
view hashes)
Built Distribution
clax-0.0.6-py3-none-any.whl
(6.5 kB
view hashes)