Index of /open-courses-0923/02-CS224U-Natural-Language-Understanding/github/repair-google--jax/docs/
../
101/ 14-Oct-2026 11:08 -
201/ 14-Oct-2026 11:08 -
301/ 14-Oct-2026 11:08 -
401/ 14-Oct-2026 11:08 -
501/ 14-Oct-2026 11:08 -
601/ 14-Oct-2026 11:08 -
_static/ 14-Oct-2026 11:08 -
_templates/ 14-Oct-2026 11:08 -
_tutorials/ 14-Oct-2026 11:08 -
debugging/ 14-Oct-2026 11:08 -
export/ 14-Oct-2026 11:08 -
ffi/ 14-Oct-2026 11:08 -
internals/ 14-Oct-2026 11:08 -
jep/ 14-Oct-2026 11:08 -
notebooks/ 14-Oct-2026 11:08 -
pallas/ 14-Oct-2026 11:08 -
sphinxext/ 14-Oct-2026 11:08 -
README.md 14-Oct-2026 11:08 127
__init__.py 14-Oct-2026 11:08 581
about.md 14-Oct-2026 11:08 5512
accuracy.md 14-Oct-2026 11:08 14K
advanced_autodiff.md 14-Oct-2026 11:08 275
advanced_guides.rst 14-Oct-2026 11:08 478
aot.md 14-Oct-2026 11:08 10K
api_compatibility.md 14-Oct-2026 11:08 5005
array_refs.ipynb 14-Oct-2026 11:08 32K
array_refs.md 14-Oct-2026 11:08 21K
array_refs.py 14-Oct-2026 11:08 21K
async_dispatch.rst 14-Oct-2026 11:08 5409
autodidax.ipynb 14-Oct-2026 11:08 146K
autodidax.md 14-Oct-2026 11:08 104K
autodidax.py 14-Oct-2026 11:08 104K
autodidax2_part1.ipynb 14-Oct-2026 11:08 34K
autodidax2_part1.md 14-Oct-2026 11:08 19K
autodidax2_part1.py 14-Oct-2026 11:08 19K
automatic-differentiation.md 14-Oct-2026 11:08 8583
automatic-vectorization.md 14-Oct-2026 11:08 4287
beginner_guide.rst 14-Oct-2026 11:08 2066
benchmarking.md 14-Oct-2026 11:08 3329
buffer_donation.md 14-Oct-2026 11:08 3362
build_custom_gpu.sh 14-Oct-2026 11:08 1947
building_on_jax.md 14-Oct-2026 11:08 4917
changelog.md 14-Oct-2026 11:08 33
complex-differentiation.md 14-Oct-2026 11:08 7738
concurrency.rst 14-Oct-2026 11:08 1147
conf.py 14-Oct-2026 11:08 20K
config_options.rst 14-Oct-2026 11:08 1976
contributing.md 14-Oct-2026 11:08 12K
contributor_guide.rst 14-Oct-2026 11:08 554
control-flow.md 14-Oct-2026 11:08 12K
debugging.md 14-Oct-2026 11:08 8543
deprecation.md 14-Oct-2026 11:08 2472
developer.md 14-Oct-2026 11:08 30K
device_memory_profiling.md 14-Oct-2026 11:08 4983
direct_linearize_migration.md 14-Oct-2026 11:08 1476
distributed_data_loading.md 14-Oct-2026 11:08 22K
errors.rst 14-Oct-2026 11:08 817
extensions.rst 14-Oct-2026 11:08 371
external-callbacks.md 14-Oct-2026 11:08 15K
faq.rst 14-Oct-2026 11:08 15K
fault_tolerance.rst 14-Oct-2026 11:08 64K
ffi.ipynb 14-Oct-2026 11:08 41K
ffi.md 14-Oct-2026 11:08 32K
ffi.py 14-Oct-2026 11:08 32K
glossary.rst 14-Oct-2026 11:08 4587
gpu_memory_allocation.rst 14-Oct-2026 11:08 5694
gpu_performance_tips.md 14-Oct-2026 11:08 23K
gradient-checkpointing.md 14-Oct-2026 11:08 22K
hero.html 14-Oct-2026 11:08 398
higher-order.md 14-Oct-2026 11:08 15K
hijax_custom_derivatives.ipynb 14-Oct-2026 11:08 52K
hijax_custom_derivatives.md 14-Oct-2026 11:08 36K
hijax_custom_derivatives.py 14-Oct-2026 11:08 36K
hijax_types.ipynb 14-Oct-2026 11:08 60K
hijax_types.md 14-Oct-2026 11:08 42K
hijax_types.py 14-Oct-2026 11:08 43K
index.rst 14-Oct-2026 11:08 6583
installation.md 14-Oct-2026 11:08 20K
investigating_a_regression.md 14-Oct-2026 11:08 6750
jacobian-vector-products.md 14-Oct-2026 11:08 17K
jax-101.rst 14-Oct-2026 11:08 561
jax.ad_checkpoint.rst 14-Oct-2026 11:08 197
jax.debug.rst 14-Oct-2026 11:08 618
jax.distributed.rst 14-Oct-2026 11:08 196
jax.dlpack.rst 14-Oct-2026 11:08 188
jax.dtypes.rst 14-Oct-2026 11:08 267
jax.example_libraries.optimizers.rst 14-Oct-2026 11:08 195
jax.example_libraries.rst 14-Oct-2026 11:08 943
jax.example_libraries.stax.rst 14-Oct-2026 11:08 177
jax.experimental.checkify.rst 14-Oct-2026 11:08 345
jax.experimental.compilation_cache.rst 14-Oct-2026 11:08 265
jax.experimental.custom_partitioning.rst 14-Oct-2026 11:08 198
jax.experimental.jet.rst 14-Oct-2026 11:08 134
jax.experimental.key_reuse.rst 14-Oct-2026 11:08 120
jax.experimental.mesh_utils.rst 14-Oct-2026 11:08 225
jax.experimental.multihost_utils.rst 14-Oct-2026 11:08 453
jax.experimental.pallas.mosaic_gpu.rst 14-Oct-2026 11:08 1808
jax.experimental.pallas.rst 14-Oct-2026 11:08 816
jax.experimental.pallas.tpu.rst 14-Oct-2026 11:08 1352
jax.experimental.pallas.triton.rst 14-Oct-2026 11:08 483
jax.experimental.random.rst 14-Oct-2026 11:08 185
jax.experimental.rst 14-Oct-2026 11:08 730
jax.experimental.serialize_executable.rst 14-Oct-2026 11:08 241
jax.experimental.sparse.rst 14-Oct-2026 11:08 2306
jax.experimental.xla_metadata.rst 14-Oct-2026 11:08 251
jax.export.rst 14-Oct-2026 11:08 1216
jax.extend.backend.rst 14-Oct-2026 11:08 312
jax.extend.core.rst 14-Oct-2026 11:08 765
jax.extend.linear_util.rst 14-Oct-2026 11:08 320
jax.extend.lowering.rst 14-Oct-2026 11:08 200
jax.extend.mlir.rst 14-Oct-2026 11:08 271
jax.extend.pallas.rst 14-Oct-2026 11:08 176
jax.extend.random.rst 14-Oct-2026 11:08 280
jax.extend.rst 14-Oct-2026 11:08 311
jax.extend.xla.rst 14-Oct-2026 11:08 215
jax.ffi.rst 14-Oct-2026 11:08 204
jax.flatten_util.rst 14-Oct-2026 11:08 224
jax.image.rst 14-Oct-2026 11:08 332
jax.lax.rst 14-Oct-2026 11:08 5367
jax.nn.initializers.rst 14-Oct-2026 11:08 933
jax.nn.rst 14-Oct-2026 11:08 800
jax.numpy.rst 14-Oct-2026 11:08 10K
jax.ops.rst 14-Oct-2026 11:08 489
jax.profiler.rst 14-Oct-2026 11:08 723
jax.random.rst 14-Oct-2026 11:08 1091
jax.ref.rst 14-Oct-2026 11:08 258
jax.rst 14-Oct-2026 11:08 4787
jax.scipy.rst 14-Oct-2026 11:08 7535
jax.sharding.rst 14-Oct-2026 11:08 376
jax.stages.rst 14-Oct-2026 11:08 507
jax.test_util.rst 14-Oct-2026 11:08 238
jax.tree.rst 14-Oct-2026 11:08 364
jax.tree_util.rst 14-Oct-2026 11:08 900
jax.typing.rst 14-Oct-2026 11:08 178
jax_array_migration.md 14-Oct-2026 11:08 12K
jit-compilation.md 14-Oct-2026 11:08 11K
key-concepts.md 14-Oct-2026 11:08 7585
migrate_pmap.md 14-Oct-2026 11:08 35K
multi_process.md 14-Oct-2026 11:08 40K
parallel.ipynb 14-Oct-2026 11:08 31K
parallel.md 14-Oct-2026 11:08 20K
parallel.py 14-Oct-2026 11:08 20K
persistent_compilation_cache.md 14-Oct-2026 11:08 12K
profiling.md 14-Oct-2026 11:08 19K
pytrees.md 14-Oct-2026 11:08 13K
random-numbers.md 14-Oct-2026 11:08 9682
requirements.txt 14-Oct-2026 11:08 733
security.md 14-Oct-2026 11:08 5779
shardy_jax_migration.md 14-Oct-2026 11:08 6471
stateful-computations.md 14-Oct-2026 11:08 8140
superseded.rst 14-Oct-2026 11:08 1222
the-training-cookbook.py 14-Oct-2026 11:08 8988
the-training-cookbook.rst 14-Oct-2026 11:08 26K
thunky.md 14-Oct-2026 11:08 35K
tracing.md 14-Oct-2026 11:08 8157
transfer_guard.rst 14-Oct-2026 11:08 2826
whats-new.md 14-Oct-2026 11:08 2310
xla_flags.md 14-Oct-2026 11:08 2356