A collection of jax functions to help with common machine/deep learning related functionality.
Project description
jax_toolkit
A collection of jax functions to help with common machine/deep learning related functionality.
This library currently contains the basics for a number of losses and metrics. We intend to add more complexity and functionality as and when it's needed - of course contributions/pull requests/bug reports etc. are very welcome if you discover problems or need something that is currently missing.
Installation
pip install jax_toolkit
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
jax_toolkit-0.1.1.tar.gz
(12.7 kB
view hashes)
Built Distribution
Close
Hashes for jax_toolkit-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 56089b686596afee181a13bdcd4c06c90694868cc84ffedf98484201743cf6e5 |
|
MD5 | 245f0ea37a9adc79d069dfcc3935875a |
|
BLAKE2b-256 | 1ddc59c6f0791926edb46e7b920856418fdef0b8f0351a6623cb0c50b27862ac |