Metadata-Version: 2.4
Name: adv-jax-math
Version: 0.2
Summary: Advanced automatic-differentiation and batching utilities for JAX
Home-page: https://github.com/unalmis/adv-jax-math
Author: Kaya Unalmis
Author-email: kunalmis@stanford.edu
License: LGPL-3.0
Project-URL: Issues Tracker, https://github.com/unalmis/adv-jax-math/issues
Project-URL: Contributing, https://github.com/unalmis/adv-jax-math/blob/main/.github/CONTRIBUTING.rst
Project-URL: Source Code, https://github.com/unalmis/adv-jax-math/
Project-URL: Documentation, https://unalmis.github.io/adv-jax-math/
Keywords: jax automatic-differentiation batching mathematics
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Typing :: Typed
Requires-Python: >=3.10
Description-Content-Type: text/x-rst
License-File: LICENSE
Requires-Dist: equinox!=0.13.3,>=0.11.0
Requires-Dist: jax!=0.7.1,!=0.8.2,!=0.9.0,>=0.6.2
Requires-Dist: jaxtyping>=0.2.24
Requires-Dist: numpy>=1.20.0
Requires-Dist: packaging
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: keywords
Dynamic: license
Dynamic: license-file
Dynamic: project-url
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

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

|Docs| |UnitTests| |Codecov|

``adv-jax-math`` provides automatic-differentiation and batching
utilities for 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`_. Portions adapted from JAX retain their original attribution
notices in the source.

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