Name		Name	Last commit message	Last commit date
.github		.github
adv_jax_math		adv_jax_math
devtools		devtools
docs		docs
tests		tests
.gitattributes		.gitattributes
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
CITATION.cff		CITATION.cff
LICENSE		LICENSE
MANIFEST.in		MANIFEST.in
README.rst		README.rst
pyproject.toml		pyproject.toml
requirements.txt		requirements.txt
setup.cfg		setup.cfg
setup.py		setup.py
versioneer.py		versioneer.py

Repository files navigation

adv-jax-math

$Documentation$ $UnitTests$

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

Installation

adv-jax-math is installable with pip:

pip install adv-jax-math

For full details, see the API documentation.

Licensing

Copyright (C) 2024 Kaya Unalmis.

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

About

Batching and sharding utilities for vectorization and differentiation in JAX

unalmis.github.io/adv-jax-math/

LGPL-3.0 license

3 stars

0 watching

Report repository

Releases

Packages

Used by

Contributors

Languages