burn-core 0.22.0

Flexible and Comprehensive Deep Learning Framework in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

# Burn Core

> Modules, configuration, records and data loading for [Burn](https://github.com/tracel-ai/burn)

[![Current Crates.io Version](https://img.shields.io/crates/v/burn-core.svg)](https://crates.io/crates/burn-core)
[![Documentation](https://docs.rs/burn-core/badge.svg)](https://docs.rs/burn-core)
[![license](https://shields.io/badge/license-MIT%2FApache--2.0-blue)](https://github.com/tracel-ai/burn/blob/main/LICENSE-MIT)

Applications should depend on [burn](https://crates.io/crates/burn), which re-exports everything
here together with neural network layers, optimizers and training. This crate is the dependency for
libraries that only need the core abstractions.

- `tensor`: Burn's tensor API, re-exported from [burn-tensor](https://github.com/tracel-ai/burn/tree/main/crates/burn-tensor).
- `module`: the `Module` trait, implemented with `#[derive(Module)]`, and `Param` for trainable
  tensors.
- `config`: serializable configuration structs with `#[derive(Config)]`.
- `store`: module records and the burnpack format.
- `data`: datasets, batchers and data loaders (`std` only).
- `prelude`: the types most programs import.

## Feature Flags

Backend features (`wgpu`, `cuda`, `flex`, ...) and the `autodiff`, `fusion`, `remote` and `capture`
features match those of `burn`. Others:

- `std` (default): standard library support. Without it the crate is `no_std` with `alloc`.
- `dataset`: the dataset library in `data`; `vision`, `audio` and `sqlite` add sources.
- `network`: file downloads with a progress bar.
- `tracing`: instrument operations with the `tracing` crate.

<!-- burn-crate-footer -->

---

Part of the [Burn](https://github.com/tracel-ai/burn) deep learning framework. See the
[Burn Book](https://burn.dev/books/burn/) and the [API documentation](https://docs.rs/burn).