Nicolas Patry fa3ea98ba9 Adding bfloat16 support for the cast kernels. (#1520)

2024-01-04 12:12:56 +01:00

..

cpu

Add an erf based gelu op (#900)

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

quantized

Fix the quantized mistral example. (#1478)

2023-12-25 09:31:24 +01:00

accelerate.rs

Accelerate support for gelu. (#782)

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

backend.rs

Add the conv-transpose1d op. (#1251)

2023-11-03 09:44:46 +01:00

backprop.rs

Upsample grad (#1420)

2023-12-10 08:43:24 +01:00

conv.rs

Test for the transposed conv1d. (#1254)

2023-11-03 13:10:28 +01:00

convert.rs

Add support for i64 (#563)

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

cpu_backend.rs

Add the conv-transpose1d op. (#1251)

2023-11-03 09:44:46 +01:00

cuda_backend.rs

Add the conv-transpose1d op. (#1251)

2023-11-03 09:44:46 +01:00

cudnn.rs

Override the default cudnn heuristics. (#957)

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

device.rs

Implement randn (CPU-> device)

2023-12-17 19:09:08 +01:00

display.rs

Tmp state.

2023-11-20 14:12:57 +01:00

dtype.rs

Only optimize float tensors. (#1069)

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

dummy_cuda_backend.rs

Add the conv-transpose1d op. (#1251)

2023-11-03 09:44:46 +01:00

dummy_metal_backend.rs

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

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

error.rs

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

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

indexer.rs

Fix a couple typos (#1451)

2023-12-17 05:20:05 -06:00

layout.rs

Retrieve more information from PyTorch checkpoints. (#515)

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

lib.rs

Implement the module trait directly for QMatMul. (#1372)

2023-11-25 10:09:45 +00:00

metal_backend.rs

Adding bfloat16 support for the cast kernels. (#1520)

2024-01-04 12:12:56 +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

Upsample grad (#1420)

2023-12-10 08:43:24 +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

Fix lints for clippy 1.75. (#1494)

2023-12-28 20:26:20 +01:00

storage.rs

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

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

strided_index.rs

…

tensor.rs

Validate the kernel size in pooling ops. (#1473)

2023-12-23 11:19:22 +01: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

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +01:00