Some generic tools for JAX
Project description
This library provides some useful additional tools to JAX, such as a multivariate Newton method, val_and_jacfwd or vectorized JVP and VJP functions.
The module documentation can be found on Readthedocs.
Installation with pip
It’s as simple as typing
pip install grgrjaxin the terminal (Unix/Mac) or the Anaconda Prompt (Win).
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
grgrjax-0.4.6.tar.gz
(10.1 kB
view details)
Built Distribution
File details
Details for the file grgrjax-0.4.6.tar.gz.
File metadata
- Download URL: grgrjax-0.4.6.tar.gz
- Upload date:
- Size: 10.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.11.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e76d2009999f043ced42fc4aa5912f5f6789ebdd8186201869c43853cd59eff5 |
|
| MD5 | dbf0c0735d286b0f62463cdf1d858fa3 |
|
| BLAKE2b-256 | 257730b5ee0a7466127f3b67e08158313862d186b465a3ebd76c5354427b0563 |
File details
Details for the file grgrjax-0.4.6-py3-none-any.whl.
File metadata
- Download URL: grgrjax-0.4.6-py3-none-any.whl
- Upload date:
- Size: 7.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.11.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 88a6de3eb1e7c2e468d36d0ee8b79f56352ebe6e58c9562ad030a6b54c51a791 |
|
| MD5 | 88f34fe18a4b71aaeddbd583fedb6cad |
|
| BLAKE2b-256 | 17e28ad5721dc0ba829c5137ee53f30cf9073282e8997b1efc8ef1f5a88981ec |
