Functions for JAX

Project description

Jaxonfunctions

This is a collection of functions for JAX. This library uses jaxtyping and beartype for type checking and type hinting. This ensures that the functions are well-typed and that the types are checked at runtime.

Project details

Release history Release notifications | RSS feed

This version

0.1.0

May 18, 2024

Download files

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

Source Distribution

jaxonfunctions-0.1.0.tar.gz (4.2 kB view hashes)

Uploaded May 18, 2024 Source

Built Distribution

jaxonfunctions-0.1.0-py3-none-any.whl (3.9 kB view hashes)

Uploaded May 18, 2024 Python 3

Hashes for jaxonfunctions-0.1.0.tar.gz

Hashes for jaxonfunctions-0.1.0.tar.gz
Algorithm	Hash digest
SHA256	`725f8b4edbcd7f337683d820f03e1283c6af4b9e452f161046ca9f5bc3828cfa`
MD5	`c3dad6a897805dc9f635dd8d24e69721`
BLAKE2b-256	`61c7916e3d108dd9a807bc83f04c9aed87b258735249741eff38911038823188`

Hashes for jaxonfunctions-0.1.0-py3-none-any.whl

Hashes for jaxonfunctions-0.1.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`fc714c0e6cf1211bfae7c5692325fd3ed0f1edd405e82225af70620aee802be4`
MD5	`86e962da7cf147bd65ece05ab84bb86a`
BLAKE2b-256	`4456c3caa28384721c5dc70943c412da65adcb9038a345655b3d4a638b14b93e`