Laurent Mazare 2ed78ab336 Support for quantized tensors in the python api. (#706) ...

* Add more pyo3 support.

* Add some support for quantized tensors in pyo3.

* Add an arc layer on qmatmul.

* Add the quantized matmul.

* Quantization support.

* More quantization support.

* Test the python quantization.

2023-09-01 15:53:42 +01:00

..

cpu

Fix for when f16c is not available. (#614)

2023-08-27 07:19:52 +01:00

quantized

Support for quantized tensors in the python api. (#706)

2023-09-01 15:53:42 +01:00

accelerate.rs

Fix the accelerate build (#678)

2023-08-30 18:31:14 +02:00

backend.rs

Add the powf op. (#664)

2023-08-29 20:48:18 +01:00

backprop.rs

Add tanh. (#675)

2023-08-30 13:54:50 +01:00

conv.rs

Dilated convolutions (#657)

2023-08-29 16:12:11 +01:00

convert.rs

Add support for i64 (#563)

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

cpu_backend.rs

Support dilation in conv-transpose2d. (#671)

2023-08-30 09:22:00 +01:00

cuda_backend.rs

Support dilation in conv-transpose2d. (#671)

2023-08-30 09:22:00 +01:00

cudnn.rs

Dilated convolutions (#657)

2023-08-29 16:12:11 +01:00

device.rs

Cleanup the pyo3 setup. (#705)

2023-09-01 14:26:18 +01:00

display.rs

Fix the debug implementation. (#648)

2023-08-28 22:51:39 +01:00

dtype.rs

Fixes for clippy 1.72. (#587)

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

dummy_cuda_backend.rs

Add the powf op. (#664)

2023-08-29 20:48:18 +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

Simplify usage of the pool functions. (#662)

2023-08-29 19:12:16 +01:00

mkl.rs

Add tanh. (#675)

2023-08-30 13:54:50 +01:00

npy.rs

Add support for i64 (#563)

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

op.rs

Add tanh. (#675)

2023-08-30 13:54:50 +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

Add the powf op. (#664)

2023-08-29 20:48:18 +01:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

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

tensor.rs

More ops again. (#697)

2023-08-31 22:28:48 +01:00

test_utils.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +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