mirror of
https://github.com/tracel-ai/burn.git
synced 2026-05-31 19:49:48 +09:00
dbf03c516b
* Remove backend generic from `Tensor` and add high-level `Device` struct * Fix rfft * WIP modules * Cleanup flags * Carry autodiff intent via checkpointing field * Fix ctc loss test shape * Fix imports * Fix burn-collective & burn-communication * Working core / nn / optim * Removed from burn-rl * Removed from burn-train * Update lock * Working mnist * More examples * Fix int/bool to_device with float AD device + add enumerate devices * Working text-classification * Fixed MultiGradientsParams usage + added gradient checkpointing to trainer * Updated text-generation example * Update comment * Fix merge * Remove backend generics from merge * Update lock * Remove from burn-store * Fix burn-store dtype tests * Remove circular dep to burn-tensor from burn-flex * Cleanup * Fix burn-store tests * Fix seed * Fix imports * More examples * Add `backend_extension` proc macro * Working vision extension * Missing merge changes * More merge fixes * tree * Whoops * Cleanup * Not feature gated * Fix wgpu kernel example * Small change * Update lock * Cleanup * Cargo fmt * Clippy * Remove burn-collective * Cleanup * Fix burn-store import * Fallback to wgpu for workspace builds & blanket fusion impl * Add `default_backend` fallback for burn-dispatch * Gate unused var autodiff * Fix no-std * Refactor new tests * Fix more tests * Adjust ctc_loss backward f16 tolerance * Fix test device for burn-core * Fix test device * Fix some docs * Default to ndarray for now due to burn-flex CAS limitation * Fix docs + ndarray default * Add portable_atomic_util::Arc in burn-autodiff * Add optim feature flag because it requires autodiff (fix no-std; not supported on all targets) * Add Gradients high-level wrapper * Fix types * Fix test device * Fix docs * Fix burn-train auto merge * Fix clippy * WIP burn device override * Fix dtype usage * Remove test-metal * Fix CI test flags * Remove burn-train vision on metal * Small note * Missing metal feature for macos CI * Fix docs * Remove backend ops note * Change set_default_dtypes to &mut self * Fix mut device * Fix burn-vision backend features * Exclude burn-backend-tests from workspace tests (explicitly handled)