Back to packageRELEASE DETAILS

1.3

PyPI

adv-jax-math

Advanced automatic-differentiation and batching utilities for JAX

Published30 Sept 2026

Direct dependencies3

Transitive dependenciesNot calculated

README

Version 1.3 documentation

############
adv-jax-math
############
|License| |Issues| |Pypi|

|Docs| |UnitTests| |Codecov|

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

Installation
============

``adv-jax-math`` is installable with ``pip``:

.. code-block:: sh

    pip install adv-jax-math


For full details, see the `API documentation <https://unalmis.github.io/adv-jax-math/>`__.

Licensing
=========

Copyright (C) 2024 Kaya Unalmis.

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

.. _GNU Lesser General Public License, version 3:
   https://github.com/unalmis/adv-jax-math/blob/main/LICENSE


.. |License| image:: https://img.shields.io/github/license/unalmis/adv-jax-math?color=blue&logo=open-source-initiative&logoColor=white
    :target: https://github.com/unalmis/adv-jax-math/blob/main/LICENSE
    :alt: License

.. |Docs| image:: https://github.com/unalmis/adv-jax-math/actions/workflows/docs.yml/badge.svg
    :target: https://unalmis.github.io/adv-jax-math/
    :alt: Documentation

.. |UnitTests| image:: https://github.com/unalmis/adv-jax-math/actions/workflows/unittest.yml/badge.svg
    :target: https://github.com/unalmis/adv-jax-math/actions/workflows/unittest.yml
    :alt: UnitTests

.. |Codecov| image:: https://codecov.io/gh/unalmis/adv-jax-math/graph/badge.svg
    :target: https://codecov.io/gh/unalmis/adv-jax-math
    :alt: Coverage

.. |Issues| image:: https://img.shields.io/github/issues/unalmis/adv-jax-math
    :target: https://github.com/unalmis/adv-jax-math/issues
    :alt: GitHub issues

.. |Pypi| image:: https://img.shields.io/pypi/v/adv-jax-math
    :target: https://pypi.org/project/adv-jax-math/
    :alt: PyPI

Downloads & popularity

Awaiting observations

No metric observations yet

Download counts and popularity will appear here as the collector receives them.

Declared dependencies 3

Native requirements

3 direct targetsNot calculated transitive count

Package / target Requirement Relationship

Package / target	Requirement	Relationship
equinox Environment { "extras": "", "marker": null, "requires_python": ">=3.10", "evaluation": "NOT_EVALUATED" }	`equinox!=0.13.3,>=0.11.0`	runtime
jax Environment { "extras": "", "marker": null, "requires_python": ">=3.10", "evaluation": "NOT_EVALUATED" }	`jax!=0.5.1,!=0.5.2,!=0.6.0,!=0.7.1,>=0.4.38`	runtime
packaging Environment { "extras": "", "marker": null, "requires_python": ">=3.10", "evaluation": "NOT_EVALUATED" }	`packaging`	runtime

equinox

Environment

{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}

equinox!=0.13.3,>=0.11.0

runtime

jax

Environment

{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}

jax!=0.5.1,!=0.5.2,!=0.6.0,!=0.7.1,>=0.4.38

runtime

packaging

Environment

{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}

packaging

runtime

Distribution files

2 collected

adv_jax_math-1.3-py3-none-any.whl

bdist_wheel · 18.88 KB · Python py3

Source file

adv_jax_math-1.3.tar.gz

sdist · 48.27 KB · Python source

Source file

Repository

PENDING

unalmis/adv-jax-math

Shared repository data · GitHub

License detected in repository: Not collected

Metadata history

Meaningful changes

Current collected state Current

Observed since 20 Oct, 20:04 UTC

No superseded metadata states collected. Unchanged observations do not create new revisions.

Record identity

package_version:68d96b70c482e112e6533bbc4226bc615631622158a2e7f9c3c2e82ac8e4f820

Other collected versions

View all collected versions →