XuanxiZhang

Pinned Repositories

torch_ACA
repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE
Language:Python00

XuanxiZhang's Repositories

XuanxiZhang/torch_ACA
repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE