mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/examples
History
Jake VanderPlas f090074d86 Avoid 'from jax import config' imports 
In some environments this appears to import the config module rather than
the config object.
2024-04-11 13:23:27 -07:00
..
jax_cpp
[PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.
2024-04-02 17:20:16 -07:00
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
advi.py
update top-level examples to use new-style typed keys
2024-03-07 12:40:10 -08:00
datasets.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
differentially_private_sgd.py
update top-level examples to use new-style typed keys
2024-03-07 12:40:10 -08:00
examples_test.py
Avoid 'from jax import config' imports
2024-04-11 13:23:27 -07:00
gaussian_process_regression.py
Avoid 'from jax import config' imports
2024-04-11 13:23:27 -07:00
kernel_lsq.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
mnist_classifier_fromscratch.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
mnist_classifier.py
update top-level examples to use new-style typed keys
2024-03-07 12:40:10 -08:00
mnist_vae.py
update top-level examples to use new-style typed keys
2024-03-07 12:40:10 -08:00
onnx2xla.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
spmd_mnist_classifier_fromscratch.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00