burn/burn-train
Nathaniel Simard 322480b744
Feat/op fusion decorator (#939)
* WIP

* Impl backend decorator

* WIP

* WIP

* WIP

* WIP

* WIP

* WIP

* Refactor

* Handle graph single ops execution

* WIP

* Starting to get concrete

* WIP

* Fix locator

* Implement add ops

* Start implementing ops

* Add more ops

* Add more ops

* More float ops

* Almost finish float ops

* Almost done with Int

* Some fix

* Into float

* Implement bool ops

* Almost done with MVP

* Fix adaptive pooling

* Add fusion as backend

* Fix memory leak

* Fix

* WIP Doc

* Doc all ops enum

* Initial docs

* Clippy

* Clippy v2

* Fix typos

* Fix doc

* Fix feature flags

* Add missing ops

* Some cleanup

* Revert u128 id

* cosmetic fixes

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
2023-11-09 21:21:41 -05:00
..
src Feat/op fusion decorator (#939) 2023-11-09 21:21:41 -05:00
Cargo.toml Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -04:00
LICENSE-APACHE Refactor/burn core (#143) 2023-01-01 19:21:08 -05:00
LICENSE-MIT Refactor/burn core (#143) 2023-01-01 19:21:08 -05:00
README.md Refactor/burn core (#143) 2023-01-01 19:21:08 -05:00

README.md

Burn Train

This crate should be used with burn.

Current Crates.io Version license