adv-jax-math

############
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