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.

Readthedocs has some minimal documentation.

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

0.4.1

Mar 22, 2023

0.4.0

Feb 24, 2023

0.3.0

Feb 18, 2023

This version

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 Feb 18, 2023 Source

Uploaded Feb 18, 2023 Python 3

Hashes for grgrjax-0.2.0.tar.gz
Algorithm	Hash digest
SHA256	`9ada667c4146ebed589a95ba2a2eab00dc1575109a07f0e81d348f78ce80793e`
MD5	`808c1e84e0a50130a495eceb09f4f4a8`
BLAKE2b-256	`ebdec934c70500966b41b7d3fd0ee073dd4b3990a824042ddbcad0412f9eedd4`

Hashes for grgrjax-0.2.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`bc30b0790e3269001b0f7eaf385f828a46a38ed70fcac651d73caa56b42488bd`
MD5	`929c8b3613c1b6ffaeca6f7eb8481d66`
BLAKE2b-256	`285a9b6dd6ad98cf94f7a0f9c110aa96f8e2bc29942fbff91b7a49a72c2ba3bc`