wozeparrotandGitHub 0dc333cfab Promote Embedding to nn (#798) ...

* feat: promote Embedding to nn

* fix: fix failing test

* feat: add test with jit

* feat: rewrite embedding to no longer need stacked for loops

* clean+fix: don't know how that happened

2023-05-25 18:39:45 -07:00

..

external

Promote Embedding to nn (#798)

2023-05-25 18:39:45 -07:00

extra

linearizer! (#714)

2023-03-19 23:43:49 -07:00

models

Add mlperf RNN-T model (#782)

2023-05-25 00:41:21 -07:00

unit

Fix PytestCollectionWarning when running tests (#791)

2023-05-24 23:17:57 -07:00

__init__.py

All devices are equal! (#196)

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

Dockerfile

testdocker

2023-01-09 12:41:52 -08:00

test_assign.py

assign buffer reuse (#547)

2023-02-09 11:53:02 -06:00

test_conv_shapetracker.py

speed like you wouldn't believe (#626)

2023-03-02 07:49:19 -08:00

test_conv.py

Add conv binops_no_rerun test assertions (#665)

2023-03-10 19:09:48 -08:00

test_custom_function.py

fix binop, other tests failure (#723)

2023-03-22 18:15:07 -07:00

test_dtype.py

casting: support all backends and implement half (#726)

2023-03-24 09:58:03 -07:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

assertions for jit

2023-05-05 21:56:32 -07:00

test_net_speed.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

test_nn.py

Promote Embedding to nn (#798)

2023-05-25 18:39:45 -07:00

test_ops.py

Add sin/cos/tan (#794)

2023-05-25 09:04:56 -07:00

test_optim.py

remove RMSprop, nobody uses it anymore

2023-03-20 12:31:34 -07:00

test_randomness.py

Statistics test: check if distributions match torch (#769)

2023-05-07 21:43:23 -07:00

test_specific_conv.py

clean up opt (#649)

2023-03-05 20:49:36 -08:00

test_speed_v_torch.py

good changes from the M1 Tensor Core project (#730)

2023-03-29 05:11:02 +04:00

test_tensor.py

ConvNeXt + extras (#657)

2023-03-06 22:10:56 -08:00