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.8.tar.gz
(10.3 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file grgrjax-0.4.8.tar.gz.
File metadata
- Download URL: grgrjax-0.4.8.tar.gz
- Upload date:
- Size: 10.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1a6bbd6663ad1c96d6f92bf4fd8f7c81951e7b7531539d19b2985ad3a44a4676
|
|
| MD5 |
ed811d1d222f984a303b693bd90dc6f3
|
|
| BLAKE2b-256 |
2d8adf7f824149d0f244258a3a9544c6b735d63903fb608ff821478fbdedaccc
|
File details
Details for the file grgrjax-0.4.8-py3-none-any.whl.
File metadata
- Download URL: grgrjax-0.4.8-py3-none-any.whl
- Upload date:
- Size: 7.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8136de3d2c03399d0345295669255cd530c4877b1fc4d54b4f203f96c4953c15
|
|
| MD5 |
bf460f5cca2bad7ee3b8e5caf9447caf
|
|
| BLAKE2b-256 |
87c9b499bc52b6b7c5f10e1a95735e5835240f95cb9758252a366c38f5ef0ce0
|
