Revision f309b82a20358a6a6560b87f8325f73ab06c3123 authored by Jevin Jiang on 05 August 2024, 18:06:42 UTC, committed by jax authors on 06 August 2024, 06:17:11 UTC
we also emulate shuffled store using (store + shuffled load + store) for previous generations. PiperOrigin-RevId: 659612665
1 parent f255fb7
| File | Mode | Size |
|---|---|---|
| .github | ||
| benchmarks | ||
| build | ||
| cloud_tpu_colabs | ||
| docs | ||
| examples | ||
| images | ||
| jax | ||
| jax_plugins | ||
| jaxlib | ||
| tests | ||
| third_party | ||
| .bazelrc | -rw-r--r-- | 17.0 KB |
| .bazelversion | -rw-r--r-- | 6 bytes |
| .editorconfig | -rw-r--r-- | 292 bytes |
| .gitignore | -rw-r--r-- | 355 bytes |
| .pre-commit-config.yaml | -rw-r--r-- | 1.1 KB |
| .readthedocs.yml | -rw-r--r-- | 570 bytes |
| AUTHORS | -rw-r--r-- | 313 bytes |
| CHANGELOG.md | -rw-r--r-- | 134.1 KB |
| CITATION.bib | -rw-r--r-- | 408 bytes |
| CONTRIBUTING.md | -rw-r--r-- | 150 bytes |
| LICENSE | -rw-r--r-- | 11.1 KB |
| README.md | -rw-r--r-- | 20.1 KB |
| WORKSPACE | -rw-r--r-- | 1.8 KB |
| conftest.py | -rw-r--r-- | 2.5 KB |
| platform_mappings | -rw-r--r-- | 371 bytes |
| pyproject.toml | -rw-r--r-- | 4.6 KB |
| setup.py | -rw-r--r-- | 4.1 KB |
Computing file changes ...
