#deep-learning #automatic-differentiation #tensor #pytorch #machine-learning

no-std burn-tensor

Tensor library with user-friendly APIs and automatic differentiation support

25 releases (16 breaking)

Uses new Rust 2024

0.17.0 Apr 23, 2025
0.16.0 Jan 14, 2025
0.15.0 Oct 28, 2024
0.13.2 May 3, 2024
0.1.0 Jul 27, 2022

#1072 in Machine learning

Download history 2584/week @ 2025-01-09 3023/week @ 2025-01-16 3175/week @ 2025-01-23 3132/week @ 2025-01-30 3318/week @ 2025-02-06 2552/week @ 2025-02-13 5539/week @ 2025-02-20 4014/week @ 2025-02-27 4163/week @ 2025-03-06 3903/week @ 2025-03-13 3225/week @ 2025-03-20 3007/week @ 2025-03-27 2329/week @ 2025-04-03 2695/week @ 2025-04-10 4635/week @ 2025-04-17 6244/week @ 2025-04-24

16,157 downloads per month
Used in 57 crates (17 directly)

MIT/Apache

1.5MB
27K SLoC

This library provides the core abstractions required to run tensor operations with Burn. Tensors are generic over the backend to allow users to perform operations using different Backend implementations. Burn's tensors also support support auto-differentiation thanks to the AutodiffBackend trait.


Burn Tensor

Burn Tensor Library

Current Crates.io Version license

This library provides the core abstractions required to run tensor operations with Burn.

Tensors are generic over the backend to allow users to perform operations using different Backend implementations. Burn's tensors also support support auto-differentiation thanks to the AutodiffBackend trait.

Dependencies

~5–40MB
~564K SLoC