New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add jax compilation cache config #198

Merged

vivianrwu merged 1 commit into AI-Hypercomputer:main from vivianrwu:jax-comp-cache

Nov 7, 2024

Collaborator

vivianrwu commented Nov 7, 2024

Adds jax_cache support #187

root@jetstream-pytorch-server-596f4c646f-l9qxx:~# cd jax_cache/
root@jetstream-pytorch-server-596f4c646f-l9qxx:~/jax_cache# ls
jit_prefill-HASH 
jit_prefill-HASH
jit_prefill-HASH

qihqi approved these changes

View reviewed changes

vivianrwu force-pushed the jax-comp-cache branch 2 times, most recently from cda2c61 to a466ff4 Compare

November 7, 2024 17:55


          Add jax compilation cache config

6429e81

vivianrwu force-pushed the jax-comp-cache branch from a466ff4 to 6429e81 Compare

November 7, 2024 18:05

vivianrwu merged commit a12698d into AI-Hypercomputer:main

3 of 4 checks passed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet