https://github.com/google/jax
History
Tip revision: b6dd15af0e464b32967175e251e9026fdff2e175 authored by Jevin Jiang on 15 September 2023, 07:36:19 UTC
[Mosaic] Fix SMEM offset calculation and support SMEM slice with tile_strides.
Tip revision: b6dd15a
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jaxlib
plugins
tests
third_party
.bazelrc -rw-r--r-- 14.2 KB
.bazelversion -rw-r--r-- 6 bytes
.editorconfig -rw-r--r-- 292 bytes
.gitignore -rw-r--r-- 342 bytes
.pre-commit-config.yaml -rw-r--r-- 1.0 KB
.readthedocs.yml -rw-r--r-- 569 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 100.0 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-- 26.0 KB
WORKSPACE -rw-r--r-- 788 bytes
conftest.py -rw-r--r-- 2.5 KB
pyproject.toml -rw-r--r-- 3.1 KB
setup.cfg -rw-r--r-- 1.5 KB
setup.py -rw-r--r-- 5.9 KB

README.md

back to top