https://github.com/google/jax
History
Tip revision: 90e0e728bb9bd3054b157d9fc80c9055b151ddf1 authored by Tomás Longeri on 15 May 2024, 21:11:47 UTC
[Mosaic] Use implicit shape over directly using `implicit_dim()` in layout.{cc, h}
Tip revision: 90e0e72
File Mode Size
jax_cpp
__init__.py -rw-r--r-- 581 bytes
advi.py -rw-r--r-- 4.8 KB
datasets.py -rw-r--r-- 3.1 KB
differentially_private_sgd.py -rw-r--r-- 8.8 KB
examples_test.py -rw-r--r-- 2.0 KB
gaussian_process_regression.py -rw-r--r-- 4.4 KB
kernel_lsq.py -rw-r--r-- 2.5 KB
mnist_classifier.py -rw-r--r-- 3.1 KB
mnist_classifier_fromscratch.py -rw-r--r-- 3.0 KB
mnist_vae.py -rw-r--r-- 4.9 KB
onnx2xla.py -rw-r--r-- 4.7 KB
spmd_mnist_classifier_fromscratch.py -rw-r--r-- 4.9 KB

back to top