mirror of https://github.com/tracel-ai/burn.git synced 2026-05-31 19:49:48 +09:00

Files

Guillaume Lagrange e18a3c2e30 refactor(backends)!: remove associated element types & replace with device defaults (#5000 )

* Remove multi backend router

* Remove associated element types; replaced by device defaults

* Fix wgpu devices

* Fix burn-store

* Fix cubecl backend extensions + device configs

* Missing fixes

* Fix clippy

* Fix wgpu backend feature gate in dispatch

* Fix wgpu/vulkan features for xtask tests

* Fix merge

2026-05-25 09:04:44 -04:00

examples

refactor(backends)!: remove associated element types & replace with device defaults (#5000 )

2026-05-25 09:04:44 -04:00

src

[Breaking] Remove Tensor backend generic and add high-level Device struct (#4717 )

2026-05-13 10:03:12 -04:00

Cargo.toml

[Breaking] Remove Tensor backend generic and add high-level Device struct (#4717 )

2026-05-13 10:03:12 -04:00

README.md

Bump tch to 0.22.0 (#3892 )

2025-10-20 16:22:32 -04:00

README.md

Text Generation

Note

This example makes use of the HuggingFace datasets library to download the datasets. Make sure you have Python installed on your computer.

The example can be run like so:

CUDA users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
export TORCH_CUDA_VERSION=cu128
cargo run --example text-generation --release

Mac users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
cargo run --example text-generation --release