kernex

Stencil computations in JAX

66 stars

1 watching

3 forks

Language: Python

last commit: 12 months ago

Linked from 1 awesome list

jaxkernelstencil

Backlinks from these awesome lists:

n2cholas/awesome-jax