grgrjax

Some generic tools for JAX

These details have not been verified by PyPI

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.

It’s as simple as typing

pip install grgrjax

in the terminal (Unix/Mac) or the Anaconda Prompt (Win).

These details have not been verified by PyPI

0.4.6

Oct 23, 2023

0.4.5

Oct 23, 2023

0.4.4

Jun 27, 2023

0.4.3

Apr 15, 2023

This version

0.4.1

Mar 22, 2023

0.4.0

Feb 24, 2023

0.3.0

Feb 18, 2023

0.2.0

Feb 18, 2023

0.1.1

Feb 17, 2023

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Uploaded Mar 22, 2023 Source

Uploaded Mar 22, 2023 Python 3

Hashes for grgrjax-0.4.1.tar.gz
Algorithm	Hash digest
SHA256	`b4e53a612bfe83e2bcf22f24595e65e7d394bfecc07d102cdb1db1c9610d4c34`
MD5	`423698ed241414be86040bed12831c4e`
BLAKE2b-256	`9da885f21958709e53463edff42d8c4849feec1283ef78387cf3c5c10551d4d6`

Hashes for grgrjax-0.4.1-py3-none-any.whl
Algorithm	Hash digest
SHA256	`6abf3c4fa94a72d1e5324743966cee822a359f076553d0818714347251cf3c9d`
MD5	`1c0fe52cb727307f7ae6653b040c22be`
BLAKE2b-256	`4409064787df5daec46ed52cb59e5c807a0dcb09ce85701cd230e9cac4ba5460`