15 releases (breaking)
new 0.13.2 | May 3, 2024 |
---|---|
0.12.1 | Feb 1, 2024 |
0.11.1 | Dec 4, 2023 |
0.10.0 | Oct 24, 2023 |
0.3.0 | Nov 20, 2022 |
#563 in Machine learning
12,665 downloads per month
Used in 14 crates
(7 directly)
1MB
26K
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
~3.5–5MB
~88K SLoC