burn/crates/burn-train
Genna Wingert 163a2069c6
Optimize connected components (#2810)
* Add boolean and/or to bool tensors and fix version mismatch for rand

* Add tests

* Fix tch

* Fix for cubecl update

* Add missing test attribute

* Optimize NdArray into_data and other ways to make connected components faster

* Use existing into_data implementation

* Make lint `allow` because it's buggy

* `bool_into_data` should have a dtype of `BoolElem`

* Don't convert to bool in place

* Don't assert type equality for bool tests

* Fix nn/mask tests

* Fix and/or tests

* Add `to_bool` and fix `Bool::Elem`

* Fix display for bool

* Fix hardcoded `DType::Bool` uses
2025-02-17 09:51:38 -05:00
..
src Optimize connected components (#2810) 2025-02-17 09:51:38 -05:00
Cargo.toml Clean up train system metrics (#2707) 2025-01-16 12:54:04 -05:00
LICENSE-APACHE Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
LICENSE-MIT Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
README.md [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00

README.md

Burn Train

This crate should be used with burn.

Current Crates.io Version license