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
126 watching
925 forks
Language: Python
last commit: about 1 year ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,371 |
 huggingface/accelerate | A tool to simplify training and deployment of PyTorch models on various devices and configurations | 7,947 |
 donnyyou/torchcv | A comprehensive PyTorch-based framework for computer vision tasks | 2,250 |
 luolc/adabound | An optimizer that combines the benefits of Adam and SGD algorithms | 2,907 |
 metaopt/torchopt | An efficient library for differentiable optimization built on top of PyTorch. | 544 |
 cornellius-gp/gpytorch | A library for creating scalable and flexible Gaussian process models with ease | 3,580 |
 shangtongzhang/deeprl | A modularized implementation of various deep reinforcement learning algorithms in PyTorch | 3,191 |
 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,286 |
 kakaobrain/torchgpipe | A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients. | 817 |
 kaiyangzhou/deep-person-reid | A PyTorch library for training and retraining deep neural networks for person re-identification in images and videos. | 4,318 |
| pytorch/torchtitan | A native PyTorch library for large-scale language model training with distributed training capabilities | 2,615 |
 pkmital/tensorflow_tutorials | A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python | 5,639 |
 junyanz/interactive-deep-colorization | A system for automatically colorizing black and white images with user interactions. | 2,694 |
 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 |