Popular
Tutorials
FAQ
View on GitHub
Execute on Binder
中文
boundscheck-mode
branch
32-default
adagrad
arctan2-doctest
array-api-devices
array-api-version
array-equal-dep
array-literals
asarray-primitive
associative-scan-docstring
at-doc
bcoo-apply
bcoo-dot-batch
bcoo-extract-bug
bcsr-to-bcoo
bernoulli-fix
beta-test-float64
bincount-doc
block-bcoo
boundscheck-mode
broadcast-in-dim
callbacks-doc
checkify-custom-jvp
choice
cholesky
cleaner-tie-in
cleanup-imports
cloud_tpu_colabs
colab-directive
collective-permute
concrete
tag
workflow_test
test-docs
jaxlib-v0.1.43
jaxlib-v0.1.42
jaxlib-v0.1.40
jaxlib-v0.1.39
jaxlib-v0.1.38
jaxlib-v0.1.37
jaxlib-v0.1.36
jaxlib-v0.1.35
jaxlib-v0.1.34
jaxlib-v0.1.33
jaxlib-v0.1.32
jax-v0.3.17
jax-v0.1.62
jax-v0.1.61
jax-v0.1.60
jax-v0.1.59
jax-v0.1.58
jax-v0.1.55
jax-v0.1.49
jax
examples
ffi
src
jax_ffi_example
Name
..
__init__.py
cpu_examples.cc
cpu_examples.py
cuda_examples.cu
cuda_examples.py
gpu_examples.cc
gpu_examples.py
rms_norm.cc
rms_norm.py