burn/crates/burn-autodiff/src
Nathaniel Simard d67cade821
Changes how vectorization is applied during fusion (#2833)
2025-02-21 10:15:27 -05:00
..
checkpoint Remove const D generic (#2298) 2024-09-24 08:35:52 -04:00
graph New autodiff graph memory management strategy (#1698) 2024-04-26 12:25:53 -04:00
ops Add boolean and/or to bool tensors (#2802) 2025-02-14 09:53:35 -05:00
runtime verifying parent nodes' existence (#2488) 2024-11-14 08:31:58 -05:00
tests Changes how vectorization is applied during fusion (#2833) 2025-02-21 10:15:27 -05:00
backend.rs [Feat] 8-bit bool for JitBackend (#2526) 2024-11-29 09:21:13 -05:00
grads.rs Add float cast op for JIT backend (#2511) 2024-11-21 13:19:33 -05:00
lib.rs Remove precision bridge and complete `float_cast` implementation for ndarray (#2538) 2024-11-28 08:28:59 -05:00
tensor.rs Add float cast op for JIT backend (#2511) 2024-11-21 13:19:33 -05:00
utils.rs [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00