Lightweight utility functions for JAX arrays, batching, and pytrees

Navigation

Verified details

These details have been verified by PyPI
Project links
GitHub Statistics
Maintainers
Avatar for djbower from gravatar.com djbower

Unverified details

These details have not been verified by PyPI
Project links
Meta
Classifiers
Report project as malware

Project description

Jaxmod

Release 0.1.0 Python 3.11+ License: GPL v3 CI Checked with pyright bear-ified Test coverage

About

Jaxmod is a Python package that provides lightweight utility functions for JAX arrays, batching, and pytrees. It mostly builds on top of the amazing Equinox package, whilst notably incorporating structural conventions and helper functions that make JAX-based scientific programming more convenient and consistent.

Documentation

The documentation is available online, with options to download it in EPUB or PDF format:

https://jaxmod.readthedocs.io/en/latest/

Quick install

Jaxmod is a Python package that can be installed on a variety of platforms (e.g. Mac, Windows, Linux). It is recommended to install Jaxmod in a dedicated Python environment. Before installation, create and activate the environment, then run:

pip install jaxmod

Project details

Verified details

These details have been verified by PyPI
Project links
GitHub Statistics
Maintainers
Avatar for djbower from gravatar.com djbower

Unverified details

These details have not been verified by PyPI
Project links
Meta
Classifiers

Release history Release notifications | RSS feed

0.4.0

0.3.0

0.2.4

0.2.3

0.2.2

0.2.0

This version

0.1.0

Download files

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

Source Distribution

jaxmod-0.1.0.tar.gz (27.4 kB view details)

Uploaded Source

Built Distribution

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

jaxmod-0.1.0-py3-none-any.whl (20.1 kB view details)

Uploaded Python 3

File details

Details for the file jaxmod-0.1.0.tar.gz.

File metadata

  • Download URL: jaxmod-0.1.0.tar.gz
  • Upload date:
  • Size: 27.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.7.20

File hashes

Hashes for jaxmod-0.1.0.tar.gz
Algorithm Hash digest
SHA256 fda367e6fc668a256eb1c8cef20ccae05b656d738dc0213530f63dc35afe5b51
MD5 bb96d6c438250f3d983514d2651a8eba
BLAKE2b-256 eb3a041c35c669c4415b45abf6800be3b857298a3d2b9d53b960d3a9cabb30cd

See more details on using hashes here.

File details

Details for the file jaxmod-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: jaxmod-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 20.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.7.20

File hashes

Hashes for jaxmod-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 7b7197a251c8a03980c739412c8adb4f7f61953347a55557901cfc46bac94e74
MD5 aed8b486d1e0bcce0896e9752f3262cd
BLAKE2b-256 778e786443ab4014f996c7a2a288fbc952ddc1429244b3c0544de40cf3ea3045

See more details on using hashes here.

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