Matthew Johnson 1f67351f56 [shard_map] make debug_print work with shard_map, eager and jit

2023-03-08 20:38:03 -08:00

..

checkify_guide.md

Replace usage of {in|out}_axis_resources with {in|out}_shardings

2023-02-28 14:29:09 -08:00

flags.md

Change jaxdb->jdb and add option to force a backend

2022-07-29 12:51:27 -07:00

index.md

[shard_map] make debug_print work with shard_map, eager and jit

2023-03-08 20:38:03 -08:00

print_breakpoint.md

Add some info in the docs about using jax.debug.print with f-strings

2023-02-15 15:16:37 -08:00