PyTorch implementation of reverse-accurate ODE solvers for Neural ODEs
Project description
PyTorch implementation of two papers: (1) Adaptive checkpoint adjoint method for gradient estimation in Neural ODEs (2) MALI: a memory efficient and reverse accurate integrator for Neural ODEs
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
TorchDiffEqPack-0.1.0.tar.gz
(26.9 kB
view hashes)
Built Distribution
Close
Hashes for TorchDiffEqPack-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 33f106a8172deeb039a76fad28daa949c2c07660d0f91541468fa7af7ecddd79 |
|
MD5 | 251716bcdc3ea3ca132348555c513e82 |
|
BLAKE2b-256 | 3572a7af0719b3a50bc366343f6224af92e46135c932766d3254cd36412203ee |