Pinned

1. google-research/torchsde Public

   Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

   Python 984 107

2. torchtyping Public

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

   Python 971 23

3. equinox Public

   Callable PyTrees and filtered transforms => neural networks in JAX. https://docs.kidger.site/equinox/

   Python 587 32

4. NeuralCDE Public

   Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

   Python 453 56

5. diffrax Public

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

   Python 503 28

6. signatory Public

   Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

   C++ 173 19