Nicolas Patry eb24875856 Reworked affine and it works ? No idea how it's different.

2023-11-08 02:37:20 +01:00

..

cpu

Add an erf based gelu op (#900)

2023-09-19 19:54:28 +01:00

quantized

Updated everything and output a trace.

2023-11-07 21:12:42 +01:00

accelerate.rs

Accelerate support for gelu. (#782)

2023-09-08 21:58:56 +01:00

backend.rs

Make the cuda rng seedable. (#1056)

2023-10-08 09:32:36 +01:00

backprop.rs

Fix the conv2d gradient computation. (#1214)

2023-10-29 09:50:04 +00:00

conv.rs

Remove some unusued bits. (#1067)

2023-10-09 19:49:57 +01:00

convert.rs

Add support for i64 (#563)

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

cpu_backend.rs

Marian MT model (#1210)

2023-10-29 15:12:22 +00:00

cuda_backend.rs

fix ucopy for f64 tensors (#1170)

2023-10-24 17:06:03 +01:00

cudnn.rs

Override the default cudnn heuristics. (#957)

2023-09-25 10:31:53 +01:00

device.rs

More scaffolding, now need to implement matmul (for precompute_cos_sin to work).

2023-11-01 16:54:09 +01:00

display.rs

Initial setup

2023-10-31 18:09:10 +01:00

dtype.rs

Only optimize float tensors. (#1069)

2023-10-10 09:05:41 +01:00

dummy_cuda_backend.rs

Make the cuda rng seedable. (#1056)

2023-10-08 09:32:36 +01:00

dummy_metal_backend.rs

Initial setup

2023-10-31 18:09:10 +01:00

error.rs

Improve metal kernel loading and associated errors

2023-11-06 09:48:18 +01:00

indexer.rs

Tensor based indexing. (#842)

2023-09-14 07:47:07 +01:00

layout.rs

Retrieve more information from PyTorch checkpoints. (#515)

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

lib.rs

Matmul (no batch, no strided, f32, f32 only) sort of done.

2023-11-01 17:36:51 +01:00

metal_backend.rs

Reworked affine and it works ? No idea how it's different.

2023-11-08 02:37:20 +01:00

mkl.rs

Add tanh. (#675)

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

npy.rs

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

op.rs

Finished scaffolding, lots of TODOs

2023-11-02 15:32:28 +01:00

pickle.rs

Handle LongStorage in pytorch checkpoints. (#1152)

2023-10-22 18:34:36 +01:00

safetensors.rs

Add the buffered safetensor wrapper. (#948)

2023-09-23 22:57:42 +01:00

scalar.rs

Get the comparison operation to work on scalar values. (#780)

2023-09-08 20:13:29 +01:00

shape.rs

PyO3: Add equal and __richcmp__ to candle.Tensor (#1099)

2023-10-30 15:17:28 +00:00

storage.rs

Matmul (no batch, no strided, f32, f32 only) sort of done.

2023-11-01 17:36:51 +01:00

strided_index.rs

…

tensor.rs

First pass (Quantized scaffolding work done + quantized example scaffolding).

2023-11-01 15:10:11 +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

First pass (Quantized scaffolding work done + quantized example scaffolding).

2023-11-01 15:10:11 +01:00

variable.rs

…