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

duality-group / jax python

Repository URL to install this package:

Details

jax / jax / _src / lax
..
control_flow
__init__.py
ann.py
convolution.py
eigh.py
fft.py
lax.py
linalg.py
other.py
parallel.py
qdwh.py
slicing.py
special.py
stack.py
svd.py
utils.py
windowed_reductions.py