/jax_parallel

Supplementary code for https://astralord.github.io/posts/exploring-parallel-strategies-with-jax/

Primary LanguagePython

No issues in this repository yet.