* update dqn-agent example
* update device stuff + couple bugs in dispatch
* update main and some utils
* deal with the whole to_device thing for off policy
* remove test code + fix warnings
* burn toml file
* feat: add TrainingProgressLogger
* feat: add a debug implementation of TrainingProgressLogger
* feat: expose new optional progress_logger
* feat: rename trait function end_eval to end_epoch and update mnist example
* feat: add EvaluationProgressLogger trait
* fix: fix evaluation_logger override
* doc: add documentation for EvaluationProgressLogger
* feat: refactor training and evaluation logger to be more general and structured
* feat: put the new methods in full.rs pipeline and evaluation pipeline
* feat: put the new methods in full.rs pipeline and evaluation pipeline
* chore: removed boolean for FullEventTrainingProcessor
* chore: remove is_test_started from EvaluationEventPorcessor and add TestStart Event
* feat: add EndEpoch event and add TrainingProgressLogger in minimal.rs
* doc: updated doc
* doc: update documentation in pardigm.rs
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
* fix: address pr comments
* feat: update method signature from new traits
* fix: change output directory in mnist and add total_test to EvalEven::start
* feat: add EndTest event and refactor evaluation pipeline to ugrade logging
* fix: reset default values in mnist example
* feat: add new event of TrainingProgressLogger to ddp algorithm
* fix: doc test and lint
* Remove generics from tensor kind traits and Elem associated type
* Move backend extension, re-export no types from burn_backend and fix into_scalar missing types
* Remove into/from primitive usage in burn-optim
* Fix backtrace
* Fix float_sort_with_indices int dtype
* Fix feature gated
* Remove dead code
* Add tensor primitive note
* Another note
* High level kind
* Working but still pub traits
* Fix into_scalar doc example
* Fix argsort out dtype
* Restrict ops traits to pub(crate)
* Add other backends for burn-vision
Consumers now say `workspace = true` instead of repeating
`path = "../burn-X", version = "=0.21.0-pre.3"` 180+ times.
Workspace deps pin `default-features = false` because Cargo forbids
consumers from overriding `default-features` when inheriting. Consumers
that previously enabled defaults now opt in with `features = ["default"]`.
* Promote burn-flex over burn-ndarray in docs and examples
burn-ndarray stays available but is no longer referenced from any
user-facing path. All docs, book pages, examples, and notebooks now
point at burn-flex. burn-ndarray and burn-candle READMEs direct users
to burn-flex for new projects.
* Remove NdArray from remaining crate-level docs
Doc tests and module headers that still mentioned NdArray as an example
backend now point at Flex. Covers:
- burn-tensor linalg::lu doc example
- burn-router Router type doc example
- burn-dispatch available backends table (ndarray row marked legacy)
- burn-candle deprecation note migration target
* Update device.rs
* Address PR review: drop Flex generics, fix device format
- text-classification: drop Flex<ElemType, i32> generics in the flex
modules (Flex only implements Backend for its default
Flex<f32, i32>, so Flex<f16, i32> failed the Backend bound under
--features flex,f16). Also fixed the same pre-existing bug in the
ag-news-train flex module.
- onnx-import.md: use burn::backend::{Flex, flex::FlexDevice} so the
sample code matches the Cargo.toml that only declares burn.
- tensor.md, notebook outputs: tensor display formats the device
with {:?}, which renders FlexDevice (not Flex or Cpu).
* Update version to 0.21.0-pre.3
* Use published pre.3 versions
* Add missing syn feature
* Move fusion tests
* Remove pretrained losses from burn-nn
* Move GramMatrixLoss to burn-vision
* Move burn-vision out of burn-core and into root burn crate
* Temporarily disable fusion tests using burn::nn
* doc(notebook) : add more basic operations and some examples
Add execution outputs to basic-tensor-op.ipynb
- Add outputs for trigonometric, reduction and comparison operations
- Add operation summary and PyTorch API comparison
Refactor plots.ipynb for autodiff and gradient descent
- Rewrite notebook to demonstrate autodiff and gradient flow
- Add GD from scratch and linear regression examples
- Update dependencies and remove unused plotting code
* resolve review feedback
* fix(notebook): - delete conclusion as review feedback
- use tensor instead of iter
- rename file
* fix: make vulkan feature opt-in for guide example
The guide example had `burn/vulkan` in its default features, which
caused `cargo build` to fail on macOS without the Vulkan SDK installed.
Workspace feature unification forced all workspace members to build
with the spirv feature, triggering a panic in cubecl-wgpu's build script.
Move vulkan to a non-default opt-in feature, matching the pattern used
by other examples (mnist, text-classification, server).
* skip burn-cuda tests on macOS
CUDA is not available on macOS. Gate the entire test module with
cfg(not(target_os = "macos")) to avoid test failures.
* fix burn-wgpu dtype test on macOS
Without the `metal` feature flag, wgpu still uses Metal at runtime on
macOS. The fallback test branch assumed a non-Metal GPU and asserted
F64 support, which Metal doesn't provide. Add a macOS-specific branch
that matches actual Metal runtime capabilities.
* enable vulkan by default on non-macOS for guide example
Use platform-conditional dependency to keep vulkan enabled by default
on Linux/Windows where it works without extra setup, while excluding
it on macOS where the Vulkan SDK is required.
* add comment for vulkan feature in guide example
* wip burn-rl
* clean up types traits, remove paradigm trait, improve learner
* code quality
* book
* doc
* error in doc
* revert TrainStep rename and remove paradigm components
* trainstep refactor with assoc types
* fix docs
* fix warning
* comment
* remove evaluator step
* add more rl stuff and change Adaptor trait
* metrics for rl and revert adaptor change
* docs + cleanup event processor for rl
* policy vs agent refactor
* rework renderer
* policy no longer generic over env
* early version of checkpointing
* versioning
* off policy training loop
* naming consistency and loading from record
* naming
* add other backends
* strategies and configs
* env initializer and cum reward metric
* transition backend thing compiles
* mostly docs and naming
* reorganize files
* batchable trait and re-arrange some bounds
* start env runner refactor
* rework autobatcher and env_runnner episodes
* change autobatcher dynamic and tweak dqn params
* fix typos and warnings
* params tweeking
* file naming
* fix docs
* metrics api and dependency bump
* address PR comments and fix lint
* kill process when user kills training
* handle kill signal renderer
* format
* remove dqn-agent from ci tests bc of dependency issue
* Add check for wasm-bindgen installation
Check for wasm-bindgen installation and install if missing.
* Simplify build script by removing wasm-bindgen check
Removed installation check for wasm-bindgen.
