Lib.rs

› Science › Machine learning › Burn
#automatic-differentiation #deep-learning #machine-learning #data

no-std burn-autodiff

Automatic differentiation backend for the Burn framework

by Nathaniel Simard and 97 contributors

16 releases (breaking)

0.14.0 Aug 27, 2024
0.13.2 May 3, 2024
0.13.1 Apr 26, 2024
0.12.1 Feb 1, 2024
0.3.0 Nov 20, 2022

#462 in Machine learning

Download history 2732/week @ 2024-06-13 1855/week @ 2024-06-20 2679/week @ 2024-06-27 2244/week @ 2024-07-04 1534/week @ 2024-07-11 1643/week @ 2024-07-18 1557/week @ 2024-07-25 3078/week @ 2024-08-01 1869/week @ 2024-08-08 1767/week @ 2024-08-15 1677/week @ 2024-08-22 1670/week @ 2024-08-29 1467/week @ 2024-09-05 1555/week @ 2024-09-12 1390/week @ 2024-09-19 1305/week @ 2024-09-26

5,948 downloads per month
Used in 25 crates (7 directly)

MIT/Apache

1.5MB
33K SLoC

Burn Autodiff

Burn autodiff backend

Current Crates.io Version license

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
~109K SLoC