Gemfury - jax · duality-group / jax v0.4.14 - python package

duality-group / jax python

Repository URL to install this package:

Details

jax / jax
..
_src
example_libraries
experimental
extend
image
interpreters
lax
lib
nn
numpy
ops
scipy
tools
__init__.py
abstract_arrays.py
ad_checkpoint.py
api_util.py
cloud_tpu_init.py
collect_profile.py
config.py
core.py
custom_batching.py
custom_derivatives.py
custom_transpose.py
debug.py
distributed.py
dlpack.py
dtypes.py
errors.py
flatten_util.py
jaxpr_util.py
linear_util.py
monitoring.py
prng.py
profiler.py
py.typed
random.py
sharding.py
stages.py
test_util.py
tree_util.py
typing.py
util.py
version.py