Diogo 4dc8595069 simple exporting models (#1344) ...

* unified exporting

* json exporting

* ignore more

* simplified buffer export

* added dtypes

* added assert

* swift example

* fix tests

* linter

* remove whitespace

* fixed tests

* remove swift example

* remove unintended changes

* allow callable models to be used

* whitespace

* more readable json export

* name change

* whitespace

* whitespace

2023-08-01 09:35:48 -07:00

..

external

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00

extra

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00

models

Add LLaMA-2 support (#1284)

2023-07-24 17:12:02 -04:00

unit

simple exporting models (#1344)

2023-08-01 09:35:48 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

Docker fix (#1039)

2023-06-25 10:38:58 -07:00

test_assign.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_conv_shapetracker.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_conv.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_custom_function.py

flake8 (#1323)

2023-07-24 11:19:58 -04:00

test_dtype.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_helpers.py

Refactored ContextVars (#1331)

2023-07-31 15:44:46 -04:00

test_jit.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_lazybuffer.py

Fix LazyBuffer SHUFFLE_PAD_OPS to prevent invalid pad movement (#1223)

2023-07-11 15:30:35 -07:00

test_linearizer.py

dedup kernel args v2 (#1272)

2023-07-18 20:03:42 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_nn.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_ops.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_optim.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_randomness.py

rng hlops: add normal and kaiming_normal (#1378)

2023-07-31 10:37:02 -07:00

test_specific_conv.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_speed_v_torch.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_symbolic_shapetracker.py

reshape to register variable value (#1386)

2023-07-31 17:10:02 -07:00

test_tensor.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00