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.2.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

This version

0.2.0

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.2.0.tar.gz (28.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.2.0-py3-none-any.whl (22.1 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for jaxmod-0.2.0.tar.gz
Algorithm Hash digest
SHA256 c22b179448596bb831d6fb40ec656740fd896a98fc31d4872689111d5afd3c9d
MD5 bf22c8bb0b4bfabef8ca042f038879e4
BLAKE2b-256 93efd9689fd1ce5aadd899aa0f31579f8b337ee761b5af2ecf808ec52cfa350d

See more details on using hashes here.

File details

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

File metadata

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

File hashes

Hashes for jaxmod-0.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 41a4c9c92e4bf81c8fb7424c9c8997ca8509bfdd57fba21fa94de9ef04f20130
MD5 c368f8c2eaa7e59b3d1b2871e3d6987d
BLAKE2b-256 fd3cf070d6eface6b6b756eac9db929ddecf812773dd27f20dfc84cab6a8adef

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