Skip to content

Commit

Permalink
Merge pull request n2cholas#75 from ASEM000/patch-1
Browse files Browse the repository at this point in the history
Added `kernex` for differentiable stencil computation.
  • Loading branch information
n2cholas authored Mar 19, 2023
2 parents e87fc61 + 3a669d3 commit 41a7ef2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,7 @@ This section contains libraries that are well-made and useful, but have not nece
- [SPU](https://github.com/secretflow/spu) - A domain-specific compiler and runtime suite to run JAX code with MPC(Secure Multi-Party Computation). <img src="https://img.shields.io/github/stars/secretflow/spu?style=social" align="center">
- [jax-tqdm](https://github.com/jeremiecoullon/jax-tqdm) - Add a tqdm progress bar to JAX scans and loops. <img src="https://img.shields.io/github/stars/jeremiecoullon/jax-tqdm?style=social" align="center">
- [safejax](https://github.com/alvarobartt/safejax) - Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`. <img src="https://img.shields.io/github/stars/alvarobartt/safejax?style=social" align="center">
- [Kernex](https://github.com/ASEM000/kernex) - Differentiable stencil decorators in JAX.

<a name="models-and-projects" />

Expand Down

0 comments on commit 41a7ef2

Please sign in to comment.