rtqichen / torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
☆5,922Updated last month
Alternatives and similar repositories for torchdiffeq:
Users that are interested in torchdiffeq are comparing it to the libraries listed below
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,473Updated last year
- Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations☆747Updated last year
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,638Updated 4 months ago
- A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…☆1,429Updated 7 months ago
- Fast and Easy Infinite Neural Networks in Python☆2,335Updated last year
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,692Updated last month
- Pytorch implementation of Augmented Neural ODEs☆541Updated 2 years ago
- Efficiently computes derivatives of NumPy code.☆7,254Updated this week
- code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".☆639Updated 4 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,882Updated last week
- Bayesian optimization in PyTorch☆3,251Updated this week
- Differentiable convex optimization layers☆1,913Updated 5 months ago
- Awesome resources on normalizing flows.☆1,516Updated this week
- JAX-based neural network library☆3,021Updated this week
- Official implementation of "Implicit Neural Representations with Periodic Activation Functions"☆1,826Updated 9 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,527Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,107Updated last year
- A small package to create visualizations of PyTorch execution graphs☆3,346Updated 4 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,651Updated last week
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,933Updated this week
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,611Updated last week
- A pedagogical implementation of Autograd☆978Updated 4 years ago
- Code for visualizing the loss landscape of neural nets☆2,972Updated 3 years ago
- A Python toolbox for performing gradient-free optimization☆4,051Updated last week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,039Updated last year
- Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)☆649Updated 2 years ago
- AI conference deadline countdowns☆5,820Updated 7 months ago
- Hopfield Networks is All You Need☆1,796Updated 2 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,091Updated last week
- Experiments with Deep Learning☆1,349Updated 2 years ago