Laurent Mazare ad33715c61 Preliminary support for importing PyTorch weights. (#511) ...

* Pickle work-in-progress.

* More unpickling.

* More pickling.

* Proper handling of setitems.

* Clippy.

* Again more pickling.

* Restore the example.

* Add enough pickle support to get the list of tensors.

* Read the data from zip files.

* Retrieve the tensor shape.

* Extract the size and dtype.

* More storage types.

* Improve the destructuring.

2023-08-19 11:26:32 +01:00

..

basics.rs

Add some conv1d test + bugfix using padding. (#349)

2023-08-08 20:50:20 +01:00

cpu_benchmarks.rs

Tensor -> QTensor conversion (#496)

2023-08-18 08:19:20 +01:00

cuda_basics.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

cuda_sum_benchmark.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00

tensor-tools.rs

Preliminary support for importing PyTorch weights. (#511)

2023-08-19 11:26:32 +01:00