Nicolas Patry dfd624dbd3 [Proposal] Remove SafeTensor wrapper (allows finer control for users).

2023-07-19 16:25:44 +02:00

..

backend.rs

Implement the backend trait for the cpu backend. (#143)

2023-07-12 09:54:33 +01:00

backprop.rs

Use the same default as pytorch for sum. (#164)

2023-07-13 21:32:32 +01:00

conv.rs

Modular backends (#138)

2023-07-11 11:17:02 +01:00

convert.rs

Add the SGD optimizer (#160)

2023-07-13 19:05:44 +01:00

cpu_backend.rs

Sum over more dims. (#197)

2023-07-19 06:46:32 +01:00

cuda_backend.rs

Revert "Testing fmt CI check behind cuda feature flag."

2023-07-14 15:18:14 +00:00

device.rs

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

display.rs

Remove some dead-code pragmas. (#137)

2023-07-11 09:33:59 +01:00

dtype.rs

Add backtrace information to errors where relevant. (#166)

2023-07-14 09:31:25 +01:00

dummy_cuda_backend.rs

Expliciting how to enable cuda.

2023-07-14 17:08:05 +02:00

error.rs

Add backtrace information to errors where relevant. (#166)

2023-07-14 09:31:25 +01:00

indexer.rs

Test the index op + bugfix. (#148)

2023-07-12 15:42:36 +01:00

layout.rs

Broadcasting performance optimization (cpu) (#182)

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

lib.rs

Broadcasting performance optimization (cpu) (#182)

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

mkl.rs

Mklize more unary ops. (#191)

2023-07-18 13:32:49 +01:00

npy.rs

Allow for lazy loading of npz files, use it in llama to reduce memory usage in the cpu version. (#141)

2023-07-11 20:22:34 +01:00

op.rs

Mklize more unary ops. (#191)

2023-07-18 13:32:49 +01:00

safetensors.rs

[Proposal] Remove SafeTensor wrapper (allows finer control for users).

2023-07-19 16:25:44 +02:00

shape.rs

Add Shape try into (#189)

2023-07-18 10:52:16 +01:00

storage.rs

Add backtrace information to errors where relevant. (#166)

2023-07-14 09:31:25 +01:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

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

tensor.rs

Iteration over strided blocks (#175)

2023-07-15 21:30:35 +01:00

utils.rs

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

variable.rs

Add backtrace information to errors where relevant. (#166)

2023-07-14 09:31:25 +01:00