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.1.tar.gz (49.3 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.1-py3-none-any.whl (19.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: adv_jax_math-1.1.tar.gz
  • Upload date:
  • Size: 49.3 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.1.tar.gz
Algorithm Hash digest
SHA256 460dd66eb378e334b66118c67f6b49c0f964aa5efd4617776cac80407d9491e5
MD5 f44f73e404021de1b23fed833a08556d
BLAKE2b-256 6ab23ab588f7ddc89129ff6686c698b9424ba111bc926f5aaa04c304bcc2ab26

See more details on using hashes here.

Provenance

The following attestation bundles were made for adv_jax_math-1.1.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.1-py3-none-any.whl.

File metadata

  • Download URL: adv_jax_math-1.1-py3-none-any.whl
  • Upload date:
  • Size: 19.3 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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 9eb4aaf3563c3bb3fdaca3c0ab94326ae178e59faaa764777f60c2645a3f39ad
MD5 1f35d4a37e2029444850c5db4d96b46d
BLAKE2b-256 ad97f861e1a6d3dced90d66515c39499899cb190f48a42b41bc227eed3955caf

See more details on using hashes here.

Provenance

The following attestation bundles were made for adv_jax_math-1.1-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