DiffEqML/torchdyn
A PyTorch library for constructing and training neural differential equation models and numerical deep learning methods.

Velocity · 7d
+0.7
★ / day
Trend
→steady
star history
Torchdyn provides utilities, layers, and functional APIs to build numerical deep learning models using neural differential equations (NeuralODE, NeuralSDE), implicit models, and GPU-compatible numerical methods. It integrates seamlessly with any PyTorch modules for composing composite models and includes tutorials and benchmarks for the deep learning research community.