torchtyping in typeddjango/awesome-python-typing

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

updated at Nov. 13, 2024, 11:56 p.m.

16 +0

1,406 +3

34 +0

diffrax in n2cholas/awesome-jax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

updated at Nov. 14, 2024, 9:37 p.m.

21 +0

1,440 +9

130 +2

equinox in n2cholas/awesome-jax

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

updated at Nov. 15, 2024, 7:58 a.m.

24 +0

2,115 +11

142 +0