Laurent Mazare ed7b99f525 Add a toggle for F16/BF16 accumulation in gemm. (#2141) ...

* Add a toggle to control f16/bf16 gemm precision.

* Use the faster variant in the quantized example.

* Bugfix.

2024-04-29 09:21:07 +02:00

..

bert

…

bigcode

…

blip

…

chatglm

…

clip

CLIP model implementation with example (#1950)

2024-03-28 13:44:12 +01:00

convmixer

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

convnext

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

custom-ops

…

dinov2

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

distilbert

…

efficientnet

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

efficientvit

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

encodec

Microphone support for the encodec example. (#1866)

2024-03-18 11:19:46 +01:00

falcon

…

gemma

Add the code-gemma models. (#2038)

2024-04-10 21:19:21 +02:00

jina-bert

…

llama

Better time measurement for the llama example. (#2106)

2024-04-22 17:54:27 +02:00

llama2-c

…

llama_multiprocess

Small cleanups to the llama multi-process example. (#2098)

2024-04-20 22:19:46 +02:00

mamba

Support alternative dtypes for mamba (#2036)

2024-04-10 18:10:01 +02:00

mamba-minimal

…

marian-mt

…

metavoice

…

mistral

Include topk sampling in the quantized example. (#2005)

2024-04-04 09:27:54 +02:00

mixtral

…

mnist-training

…

mobileone

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

moondream

Optimize copy-2d for metal. (#2024)

2024-04-07 12:34:16 +02:00

musicgen

…

olmo

Add Olmo models (#2127)

2024-04-26 11:02:51 +02:00

onnx

…

phi

Make the dtype configurable for phi. (#2133)

2024-04-27 21:32:49 +02:00

quantized

Add a toggle for F16/BF16 accumulation in gemm. (#2141)

2024-04-29 09:21:07 +02:00

quantized-phi

Add the phi-v3 quantized model. (#2118)

2024-04-24 08:22:23 +02:00

quantized-t5

Move the tensor-tools binary in a separate crate. (#1969)

2024-03-30 15:49:37 +01:00

qwen

Readme fix. (#1961)

2024-03-28 23:24:46 +01:00

recurrent-gemma

Add a quantized version of recurrent-gemma. (#2054)

2024-04-13 20:07:01 +02:00

reinforcement-learning

Update for pyo3 0.21. (#1985)

2024-04-01 17:07:02 +02:00

replit-code

…

repvgg

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

resnet

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

rwkv

…

segformer

Update the image crate + use the re-exported version. (#1893)

2024-03-21 10:56:41 +01:00

segment-anything

…

stable-diffusion

Utilize batches in Stable Diffusion (#2071)

2024-04-16 06:49:04 +02:00

stable-lm

Avoid copying the data on squeeze and unsqueeze. (#1884)

2024-03-20 13:04:36 +01:00

starcoder2

…

t5

Expose the t5 config fields + allow t5-large. (#1987)

2024-04-01 20:58:34 +02:00

trocr

Move image tensor to device in trocr example (#2063)

2024-04-15 06:50:32 +02:00

vgg

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

vit

Move the image tensor to the appropriate device. (#1856)

2024-03-16 22:25:46 +01:00

whisper

whisper readme (#1899)

2024-03-21 12:54:09 +01:00

whisper-microphone

…

wuerstchen

…

yi

…

yolo-v3

…

yolo-v8

Support alternative dtypes for mamba (#2036)

2024-04-10 18:10:01 +02:00

onnx_basics.rs

…