mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00

Files

Laurent Mazare a1812f934f Add a yolo-v3 example. (#528 )

* Add a couple functions required for yolo.

* Add the yolo-v3 example.

* Add minimum and maximum.

* Use the newly introduced maximum.

* Cuda support for min/max + add some testing.

* Allow for more tests to work with accelerate.

* Fix a typo.

2023-08-20 18:19:37 +01:00

src

Add a yolo-v3 example. (#528 )

2023-08-20 18:19:37 +01:00

build.rs

Revert "Add the layer norm files. (#222 )" (#223 )

2023-07-22 16:51:11 +01:00

Cargo.toml

Bump the crates version to 0.1.2. (#522 )

2023-08-20 08:07:07 +01:00

README.md

Revert "Add the layer norm files. (#222 )" (#223 )

2023-07-22 16:51:11 +01:00

README.md

candle-kernels

This crate contains CUDA kernels used from candle. Some of these implementations come from the dfdx crate.