zachcp 3769206583 Update docs (#2553) ...

* add module docs for candle-core

* doc each of the candle-nn modules and add the links to the doc page

2024-11-11 22:13:52 +01:00

..

cpu

Bump the version number to 0.5.1. (#2155)

2024-05-03 11:17:05 +02:00

cpu_backend

Fix the fast bf16 gemm cublas kernels. (#2274)

2024-06-18 23:46:58 +02:00

cuda_backend

Support for UG kernels. (#2579)

2024-10-27 13:37:19 +01:00

metal_backend

UG metal integration. (#2580)

2024-10-27 15:20:37 +01:00

quantized

Cuda quantized mmv bugfix. (#2526)

2024-10-01 12:57:55 +02:00

accelerate.rs

feat: add silu activation function (#1706)

2024-02-14 10:27:22 +01:00

backend.rs

Add StorageRef. (#2113)

2024-04-23 13:23:27 +02:00

backprop.rs

optimize gradient for silu a bit (#2393)

2024-08-04 11:24:17 +02:00

conv.rs

Support for groups in conv-transpose1d. (#1731)

2024-02-18 21:28:07 +01:00

convert.rs

Add support for i64 (#563)

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

custom_op.rs

UG metal integration. (#2580)

2024-10-27 15:20:37 +01:00

device.rs

UG metal integration. (#2580)

2024-10-27 15:20:37 +01:00

display.rs

Expose more printer options. (#1817)

2024-03-08 15:04:18 +01:00

dtype.rs

Add StorageRef. (#2113)

2024-04-23 13:23:27 +02:00

dummy_cuda_backend.rs

Add support for cuda streams. (#2532)

2024-10-02 21:30:58 +02:00

dummy_metal_backend.rs

Add StorageRef. (#2113)

2024-04-23 13:23:27 +02:00

error.rs

Support for UG kernels. (#2579)

2024-10-27 13:37:19 +01:00

indexer.rs

Complete the missing backticks in the comments (#2469)

2024-09-11 16:37:05 +02:00

layout.rs

Add some fast Metal MLX SDPA kernels (#2584)

2024-11-05 09:28:00 +01:00

lib.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

mkl.rs

feat: add silu activation function (#1706)

2024-02-14 10:27:22 +01:00

npy.rs

Update zip requirement from 0.6.6 to 1.1.1 (#2103)

2024-04-22 16:23:27 +02:00

op.rs

Add support for "sign" on tensors (#2012)

2024-04-04 22:32:47 +02:00

pickle.rs

Fix float unpickling. (#1730)

2024-02-18 19:33:55 +01:00

safetensors.rs

Add SliceSafetensors. (#2179)

2024-05-11 13:15:42 +02:00

scalar.rs

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

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

shape.rs

Add some fast Metal MLX SDPA kernels (#2584)

2024-11-05 09:28:00 +01:00

sort.rs

Separate quantized phi-3 implementation. (#2157)

2024-05-04 10:14:57 +02:00

storage.rs

Really unique identifier for metal device ids. (#1932)

2024-03-25 11:48:16 +01:00

streaming.rs

Stream tensor (#2429)

2024-08-17 21:54:28 +02:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

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

tensor_cat.rs

Add a slice_set op. (#2193)

2024-05-18 15:58:18 +02:00

tensor.rs

Testcases (#2567)

2024-10-17 13:00:45 +02:00

test_utils.rs

Adding the test scaffolding.

2023-11-20 14:38:35 +01:00

utils.rs

Metal part 1 - Scaffolding for metal. (#1308)

2023-11-10 08:35:48 +01:00

variable.rs

Bug Fix: When converting a tensor to a variable, clone if the tensor is already a variable. (#2124)

2024-04-29 11:21:53 +02:00