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.7.tar.gz
(7.0 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.7.tar.gz.
File metadata
- Download URL: grgrjax-0.4.7.tar.gz
- Upload date:
- Size: 7.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.13.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8eb7c3e1364e89c3a124ddbbc2dae696682e84ef1fd9e441360efe3b13a959cd
|
|
| MD5 |
6dee825bb1541d5753bd7e2fae098212
|
|
| BLAKE2b-256 |
8b5411846ff28c7a6a9efad11d74e2286719b9fdcd7a7313f1303e3414a73242
|
File details
Details for the file grgrjax-0.4.7-py3-none-any.whl.
File metadata
- Download URL: grgrjax-0.4.7-py3-none-any.whl
- Upload date:
- Size: 7.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.13.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c884da9860181b68200302d70204a06de6ffe6200e969634aedadf1d2a649bc0
|
|
| MD5 |
879d8b4c109528e8b543249902bd9fe0
|
|
| BLAKE2b-256 |
ad6a0ac87413f13e59a4ae51d5911ba3267c26df805fe3340eef324f011aa7e1
|
