Laurent Mazare 27996a1a9e Remove the old MFA gemm kernels. (#2742) ...

* Remove the old MFA gemm kernels.

* Use bf16 in helium on metal.

2025-01-26 20:36:31 +01:00

..

cpu

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

cpu_backend

Lint fixes introduced with Rust 1.83 (#2646)

2024-11-28 23:00:21 +01:00

cuda_backend

Clippy fixes for the cuda feature. (#2650)

2024-11-29 09:01:34 +01:00

metal_backend

Remove the old MFA gemm kernels. (#2742)

2025-01-26 20:36:31 +01:00

quantized

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

accelerate.rs

feat: add silu activation function (#1706)

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

backend.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

backprop.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

conv.rs

20241118 docs (#2629)

2024-11-19 04:07: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

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

display.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

dtype.rs

Add StorageRef. (#2113)

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

dummy_cuda_backend.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

dummy_metal_backend.rs

Add StorageRef. (#2113)

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

error.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

indexer.rs

Complete the missing backticks in the comments (#2469)

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

layout.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

lib.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +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

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

pickle.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

safetensors.rs

Lint fixes introduced with Rust 1.83 (#2646)

2024-11-28 23:00:21 +01:00

scalar.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

shape.rs

Add some fast Metal MLX SDPA kernels (#2584)

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

sort.rs

Fix a cuda warning. (#2693)

2024-12-31 09:06:10 +01:00

storage.rs

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

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

streaming.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

strided_index.rs

Clippy fixes for 1.84. (#2710)

2025-01-10 10:15:15 +01:00

tensor_cat.rs

Explicit error when slice-set is called with the same src and dst. (#2733)

2025-01-22 21:31:49 +01:00

tensor.rs

fix typo (#2606)

2024-11-23 08:40:00 +01:00

test_utils.rs

Adding the test scaffolding.

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

utils.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +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