equinox in n2cholas/awesome-jax

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

updated at Sept. 22, 2024, 1:03 p.m.

23 +0

2,048 +10

135 +0

diffrax in n2cholas/awesome-jax

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

updated at Sept. 21, 2024, 11:09 a.m.

20 +0

1,378 +6

124 +0

torchtyping in typeddjango/awesome-python-typing

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

updated at Sept. 20, 2024, 8:33 p.m.

16 +0

1,384 +2

33 +0