Patrick von Platen 71518caeee Align tensor device print more with PyTorch (#590) ...

* Improve tensor print

* Use CudaDevice only if enabled with cuda feature

* run rust fmt

* up

* improve

* rustfmt

2023-08-26 11:20:22 +01:00

..

cpu

Trace softmax (#568)

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

quantized

Avoid some transmutes. (#607)

2023-08-25 18:21:37 +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

Fail on bf16. (#594)

2023-08-25 06:10:38 +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

Align tensor device print more with PyTorch (#590)

2023-08-26 11:20:22 +01:00

dtype.rs

Fixes for clippy 1.72. (#587)

2023-08-24 17:46:17 +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

Neon intrinsics for the q8_0 vecdot. (#604)

2023-08-25 14:42:18 +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

Fixes for clippy 1.72. (#587)

2023-08-24 17:46:17 +01:00

pickle.rs

More pickle support. (#588)

2023-08-24 18:45:10 +01:00

safetensors.rs

Add support for i64 (#563)

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

shape.rs

Fixes for clippy 1.72. (#587)

2023-08-24 17:46:17 +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

Fixes for clippy 1.72. (#587)

2023-08-24 17:46:17 +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