Kernex
MITPythonv0.2.1
Kernex extends JAX with kmap and kscan for differentiable stencil computations, enabling efficient array transformations.
72 stars4 forks0 contributors
Overview
Stencil computations in JAX
Related Projects
Community-curated · Updated weekly · 100% open source
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
