17 releases (breaking)
0.15.0 | Oct 28, 2024 |
---|---|
0.13.2 | May 3, 2024 |
0.12.1 | Feb 1, 2024 |
0.11.1 | Dec 4, 2023 |
0.3.0 | Nov 20, 2022 |
#699 in Machine learning
9,139 downloads per month
Used in 33 crates
(7 directly)
2MB
40K
SLoC
Burn Autodiff
Burn autodiff backend
For now only first order reverse mode autodiff is supported.
lib.rs
:
Burn Autodiff
This autodiff library is a part of the Burn project. It is a standalone crate
that can be used to perform automatic differentiation on tensors. It is
designed to be used with the Burn Tensor crate, but it can be used with any
tensor library that implements the Backend
trait.
Dependencies
~6MB
~103K SLoC