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
Or for additional loss function utils:
pip install jax_toolkit[losses_utils]
1 maintainer
Metadata
1 maintainer
Release files for jax-toolkit 0.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jax_toolkit-0.2.0.tar.gz | 13.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax_toolkit-0.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 32.0 kB
Release files / jax_toolkit-0.2.0.tar.gz
| Download URL | jax_toolkit-0.2.0.tar.gz |
|---|---|
| Size | 13.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
1717977bbda0c9c275b7b4e4be01d1d500e847088ddc4ce7f44608adb82e02e1
|
|
BLAKE2b-256 checksum How to use checksums |
9bef694a7f66fc1f70618984f3e95d9852b5c8019b6ddc8737a46988888b0e3f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.2.0 pkginfo/1.6.1 requests/2.25.1 setuptools/49.2.1 requests-toolbelt/0.9.1 tqdm/4.54.1 CPython/3.9.1
|
Release files / jax_toolkit-0.2.0-py3-none-any.whl
| Download URL | jax_toolkit-0.2.0-py3-none-any.whl |
|---|---|
| Size | 18.6 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
9beaf04d608bec873e989968510426180f76d819e257bf114a4295ddcd0c9546
|
|
BLAKE2b-256 checksum How to use checksums |
7fc0eb5d3017885aa675817ff74f939af37917e3bb6945a31936e1a3aedfb6e3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.2.0 pkginfo/1.6.1 requests/2.25.1 setuptools/49.2.1 requests-toolbelt/0.9.1 tqdm/4.54.1 CPython/3.9.1
|
