torchdiffeq
by rtqichen
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
AI summary
ODE solver library
A PyTorch implementation of differentiable ODE solvers with GPU support and efficient backpropagation
- stars
- 5.7K
- forks
- 938
- watching
- 127
Similar projects
Found by comparing what the projects do, not just their names.
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
Model trainer
A tool to simplify training and deployment of PyTorch models on various devices and configurations
donnyyou/torchcv2.2K
CV framework
A comprehensive PyTorch-based framework for computer vision tasks
luolc/adabound2.9K
Optimizer
An optimizer that combines the benefits of Adam and SGD algorithms
Optimization library
An efficient library for differentiable optimization built on top of PyTorch.
GP framework
A library for creating scalable and flexible Gaussian process models with ease
Reinforcement Learning Library
A modularized implementation of various deep reinforcement learning algorithms in PyTorch
Deep Learning Tutorials
A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.
Pipeline Parallelism Library
A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients.
Person Re-Identification Library
A PyTorch library for training and retraining deep neural networks for person re-identification in images and videos.
LLM trainer
A native PyTorch library for training large language models using distributed parallelism and optimization techniques.
TensorFlow Tutorials
A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python
Image Colorizer
A system for automatically colorizing black and white images with user interactions.
ODE simulator
An ODE solver library that provides a simple API for simulating complex systems with interactive modification of differential equations.
Differentiable optimizer
A PyTorch module that wraps the OSQP solver for differentiable optimization problems
