Laurent Mazare dd64465899 Add a test for conv2d with padding + bugfix the random number generation on cuda. (#578) ...

* Add a test for conv2d with padding.

* Cosmetic changes.

* Bugfix the rand function on the cuda backend.

2023-08-24 10:16:37 +01:00

..

cpu

Trace softmax (#568)

2023-08-23 15:25:50 +01:00

quantized

GGUF support in the quantized model. (#559)

2023-08-23 09:20:57 +01:00

accelerate.rs

Add a yolo-v3 example. (#528)

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

backend.rs

add max_pool2d (#371)

2023-08-09 18:05:26 +01:00

backprop.rs

Fix the minimum/maximum gradient computations. (#534)

2023-08-21 08:28:41 +01:00

conv.rs

Add Efficientnet (#572)

2023-08-23 18:02:58 +01:00

convert.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

cpu_backend.rs

Trace softmax (#568)

2023-08-23 15:25:50 +01:00

cuda_backend.rs

Add a test for conv2d with padding + bugfix the random number generation on cuda. (#578)

2023-08-24 10:16:37 +01:00

cudnn.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

device.rs

Transpose the weight matrixes for llama2.c. (#321)

2023-08-04 13:32:20 +01:00

display.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

dtype.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

dummy_cuda_backend.rs

add max_pool2d (#371)

2023-08-09 18:05:26 +01:00

error.rs

Get the ggml based llama to generate some text. (#464)

2023-08-16 12:41:07 +01:00

indexer.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

layout.rs

Retrieve more information from PyTorch checkpoints. (#515)

2023-08-19 15:05:34 +01:00

lib.rs

Preliminary support for importing PyTorch weights. (#511)

2023-08-19 11:26:32 +01:00

mkl.rs

Add a yolo-v3 example. (#528)

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

npy.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

op.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

pickle.rs

Small tweaks to the pickle handling to be able to use libtorch files. (#530)

2023-08-20 23:25:34 +01:00

safetensors.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

shape.rs

Add a broadcast variant to matmul. (#523)

2023-08-20 13:20:42 +01:00

storage.rs

Relax the requirements on CustomOp. (#486)

2023-08-17 11:12:05 +01:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

2023-07-17 13:41:09 +01:00

tensor.rs

Add some group parameter to convolutions. (#566)

2023-08-23 12:58:55 +01:00

utils.rs

Q6K quantization (#495)

2023-08-17 22:22:57 +01:00

variable.rs

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +01:00