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 |
#752 in Machine learning
6,437 downloads per month
Used in 31 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