torchtyping in typeddjango/awesome-python-typing

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

created at March 28, 2021, 6:19 p.m.

Python

16 +0

1,345 +0

31 +0

GitHub
diffrax in n2cholas/awesome-jax

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

created at July 25, 2021, 11:42 p.m.

Python

19 +0

1,292 +8

117 +1

GitHub
equinox in n2cholas/awesome-jax

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

created at July 29, 2021, 2:21 a.m.

Python

23 +0

1,884 +8

130 +0

GitHub