mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 02:38:10 +00:00
648596c073
* added chatGLM readme * changed wording in readme * added readme for chinese-clip * added readme for convmixer * added readme for custom ops * added readme for efficientnet * added readme for llama * added readme to mnist-training * added readme to musicgen * added readme to quantized-phi * added readme to starcoder2 * added readme to whisper-microphone * added readme to yi * added readme to yolo-v3 * added readme to whisper-microphone * added space to example in glm4 readme * fixed mamba example readme to run mamba instead of mamba-minimal * removed slash escape character * changed moondream image to yolo-v8 example image * added procedure for making the reinforcement-learning example work with a virtual environment on my machine * added simple one line summaries to the example readmes without * changed non-existant image to yolo example's bike.jpg * added backslash to sam command * removed trailing - from siglip * added SoX to silero-vad example readme * replaced procedure for uv on mac with warning that uv isn't currently compatible with pyo3 * added example to falcon readme * added --which arg to stella-en-v5 readme * fixed image path in vgg readme * fixed the image path in the vit readme * Update README.md * Update README.md * Update README.md --------- Co-authored-by: Laurent Mazare <laurent.mazare@gmail.com>
17 lines
532 B
Markdown
17 lines
532 B
Markdown
# candle-custom-ops
|
|
|
|
This example illustrates how to implement forward and backward passes for custom operations on the CPU and GPU.
|
|
The custom op in this example implements RMS normalization for the CPU and CUDA.
|
|
|
|
## Running an example
|
|
|
|
```bash
|
|
$ cargo run --example custom-ops
|
|
|
|
> [[ 0., 1., 2., 3., 4., 5., 6.],
|
|
> [ 7., 8., 9., 10., 11., 12., 13.]]
|
|
> Tensor[[2, 7], f32]
|
|
> [[0.0000, 0.2773, 0.5547, 0.8320, 1.1094, 1.3867, 1.6641],
|
|
> [0.6864, 0.7845, 0.8825, 0.9806, 1.0786, 1.1767, 1.2748]]
|
|
> Tensor[[2, 7], f32]
|
|
``` |