Skip to main content

Advanced automatic-differentiation and batching utilities for JAX

Project description

License GitHub issues PyPI

Documentation UnitTests Coverage

adv-jax-math provides batching and sharding utilities for vectorization and differentiation in JAX.

Installation

adv-jax-math is installable with pip:

pip install adv-jax-math

For full details, see the API documentation.

Licensing

Copyright (C) 2024 Kaya Unalmis.

adv-jax-math is licensed under the GNU Lesser General Public License, version 3.

Project details


Download files

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

Source Distribution

adv_jax_math-1.0.tar.gz (47.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

adv_jax_math-1.0-py3-none-any.whl (19.6 kB view details)

Uploaded Python 3

File details

Details for the file adv_jax_math-1.0.tar.gz.

File metadata

  • Download URL: adv_jax_math-1.0.tar.gz
  • Upload date:
  • Size: 47.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for adv_jax_math-1.0.tar.gz
Algorithm Hash digest
SHA256 c9dbf8fcfce2a74505dcf7329d74798dedd82efa45fdb1b22041c11504d3c01f
MD5 6896db07ce8876a4856d607887c66c1e
BLAKE2b-256 62d7b58397c0583ace8f972dc3ee91021e64bcc146388581ec5495f6d5b73724

See more details on using hashes here.

Provenance

The following attestation bundles were made for adv_jax_math-1.0.tar.gz:

Publisher: release.yml on unalmis/adv-jax-math

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file adv_jax_math-1.0-py3-none-any.whl.

File metadata

  • Download URL: adv_jax_math-1.0-py3-none-any.whl
  • Upload date:
  • Size: 19.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for adv_jax_math-1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 07ea37f44a580323c93528143ec3a4101136f210bf6b59fb7fd9bc604d69a33b
MD5 20fc8b601be47d7e3e25a29e361c90b3
BLAKE2b-256 d929da6c36d986af64be9d1380c0a8f85a8b94fffbf4e8a2700b2b55f5f33cf0

See more details on using hashes here.

Provenance

The following attestation bundles were made for adv_jax_math-1.0-py3-none-any.whl:

Publisher: release.yml on unalmis/adv-jax-math

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page