candle/candle-core
Laurent Mazare f135b7963d
Fix for the batch dim in the quantized matmul example. (#2073)
* Fix for the batch dim in the quantized matmul example.

* Enable more tests on cuda.

* Add a test for qmm with a batch.

* Fix the zeros-dim test on metal.
2024-04-15 20:00:28 +02:00
..
benches Add benchmarks for qmatmul operations (#2048) 2024-04-13 12:30:14 +02:00
examples Move the tensor-tools binary in a separate crate. (#1969) 2024-03-30 15:49:37 +01:00
src Fix for the batch dim in the quantized matmul example. (#2073) 2024-04-15 20:00:28 +02:00
tests Fix for the batch dim in the quantized matmul example. (#2073) 2024-04-15 20:00:28 +02:00
Cargo.toml feat(bf16): add cast support + tests for cast + bin ops (#1524) 2024-01-11 15:49:13 +01:00
LICENSE Refactor the hierarchy. 2023-06-27 11:57:27 +02:00
README.md Refactor the hierarchy. 2023-06-27 11:57:27 +02:00

README.md

candle

Minimalist ML framework for Rust