burn/crates
Genna Wingert 97af8c6d28
Introduce autotuning to `conv2d` and `conv_transpose2d` with a new `im2col`/`GEMM` algorithm (#2287)
2024-09-23 15:54:50 -04:00
..
burn Fix CI (#2268) 2024-09-10 12:13:48 -04:00
burn-autodiff Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
burn-candle Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
burn-common Make compatible with thumbv6m-none-eabi + add raspberry pi pico example (#2096) 2024-08-23 07:39:39 -04:00
burn-core Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
burn-cuda Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-dataset Fix compile for dataset crate with vision feature (#2228) 2024-09-01 17:03:37 -04:00
burn-derive Fix module derive with generics (#2127) 2024-08-08 16:24:51 -04:00
burn-fusion Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
burn-import simplify scope tracking in burn-import (#2207) 2024-09-09 12:19:26 -04:00
burn-jit Introduce autotuning to `conv2d` and `conv_transpose2d` with a new `im2col`/`GEMM` algorithm (#2287) 2024-09-23 15:54:50 -04:00
burn-ndarray Introduce autotuning to `conv2d` and `conv_transpose2d` with a new `im2col`/`GEMM` algorithm (#2287) 2024-09-23 15:54:50 -04:00
burn-no-std-tests Refactor xtask to use tracel-xtask and refactor CI workflow (#2063) 2024-08-28 15:57:13 -04:00
burn-tch Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
burn-tensor Introduce autotuning to `conv2d` and `conv_transpose2d` with a new `im2col`/`GEMM` algorithm (#2287) 2024-09-23 15:54:50 -04:00
burn-tensor-testgen
burn-train Fix panic messages being invisible in tui mode (#2226) 2024-09-06 16:22:00 -04:00
burn-wgpu Add deform_conv2d as implemented in torchvision (#2147) 2024-09-23 15:17:23 -04:00
onnx-ir Perf/slice (#2252) 2024-09-09 11:08:39 -04:00