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-1.0.0.tar.gz
(26.1 kB
view hashes)
Built Distribution
Close
Hashes for TorchDiffEqPack-1.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 49eb0ec8ec407c264cae4fbc4a693d19d07d631859d2df2a06a98bffc74f0a5c |
|
MD5 | dd972416c837fa4caf4b948244d2df47 |
|
BLAKE2b-256 | 0901b492432cb372e4732f47fbe6319f1b8410a79bd1810787cd0bb2abeb2203 |