torchdiffeq
ODE solver library
A PyTorch implementation of differentiable ODE solvers with GPU support and efficient backpropagation
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
6k stars
127 watching
938 forks
Language: Python
last commit: almost 2 years agoRelated projects:
| Repository | Description | Stars |
|---|---|---|
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
 huggingface/accelerate | A tool to simplify training and deployment of PyTorch models on various devices and configurations | 8,056 |
 donnyyou/torchcv | A comprehensive PyTorch-based framework for computer vision tasks | 2,249 |
 luolc/adabound | An optimizer that combines the benefits of Adam and SGD algorithms | 2,908 |
 metaopt/torchopt | An efficient library for differentiable optimization built on top of PyTorch. | 554 |
 cornellius-gp/gpytorch | A library for creating scalable and flexible Gaussian process models with ease | 3,605 |
 shangtongzhang/deeprl | A modularized implementation of various deep reinforcement learning algorithms in PyTorch | 3,209 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,401 |
 kakaobrain/torchgpipe | A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients. | 820 |
 kaiyangzhou/deep-person-reid | A PyTorch library for training and retraining deep neural networks for person re-identification in images and videos. | 4,353 |
| pytorch/torchtitan | A native PyTorch library for training large language models using distributed parallelism and optimization techniques. | 2,765 |
 pkmital/tensorflow_tutorials | A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python | 5,636 |
 junyanz/interactive-deep-colorization | A system for automatically colorizing black and white images with user interactions. | 2,701 |
 soypat/godesim | An ODE solver library that provides a simple API for simulating complex systems with interactive modification of differential equations. | 23 |
 osqp/osqpth | A PyTorch module that wraps the OSQP solver for differentiable optimization problems | 59 |