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
Or for additional loss function utils:
pip install jax_toolkit[losses_utils]
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.2.0.tar.gz
(13.3 kB
view details)
Built Distribution
File details
Details for the file jax_toolkit-0.2.0.tar.gz.
File metadata
- Download URL: jax_toolkit-0.2.0.tar.gz
- Upload date:
- Size: 13.3 kB
- Tags: Source
- Uploaded using 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
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1717977bbda0c9c275b7b4e4be01d1d500e847088ddc4ce7f44608adb82e02e1 |
|
| MD5 | 21bf52a05fecbbfc3c4dbf5942af5698 |
|
| BLAKE2b-256 | 9bef694a7f66fc1f70618984f3e95d9852b5c8019b6ddc8737a46988888b0e3f |
File details
Details for the file jax_toolkit-0.2.0-py3-none-any.whl.
File metadata
- Download URL: jax_toolkit-0.2.0-py3-none-any.whl
- Upload date:
- Size: 18.6 kB
- Tags: Python 3
- Uploaded using 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
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 9beaf04d608bec873e989968510426180f76d819e257bf114a4295ddcd0c9546 |
|
| MD5 | cfbf6ed2968fe738d1b1fcffc713b691 |
|
| BLAKE2b-256 | 7fc0eb5d3017885aa675817ff74f939af37917e3bb6945a31936e1a3aedfb6e3 |
