Yixiang Gao 4d54afb6df sparse cat cross entropy (#1597) ...

* add sparse cat cross entropy

* minor fix

* add log_softmax into loss function

* add test

* update docs

* fix training loss

* add device

2023-08-21 14:14:54 -07:00

..

external

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

extra

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

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

models

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

unit

not worth the speed (#1584)

2023-08-20 10:24:58 -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_allocators.py

init allocator for compiled backends (#1467)

2023-08-17 10:33:32 -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

Shorter (#1582)

2023-08-20 08:12:16 -07:00

test_gc.py

fix up eye and fix gc test

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

test_helpers.py

symbolic codegen and exec (#1552)

2023-08-16 14:43:41 -07:00

test_jit.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_lazybuffer.py

ShapeTracker.var_vals (#1540)

2023-08-14 18:53:37 -07:00

test_linearizer.py

CSE at uop level (#1483)

2023-08-19 23:40:40 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

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

test_nn.py

sparse cat cross entropy (#1597)

2023-08-21 14:14:54 -07:00

test_ops.py

Argmax/Argmin Feature (#1576)

2023-08-20 18:46:46 -07:00

test_optim.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_randomness.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_specific_conv.py

CI < 5 minutes (#1252)

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

test_speed_v_torch.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_symbolic_jit.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_symbolic_ops.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_symbolic_shapetracker.py

more symbolic symbolic ops (#1564)

2023-08-18 09:21:41 -07:00

test_tensor.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_uops.py

make TestNonFloatUOps.test_mul_bool pass on all platforms (#1557)

2023-08-16 11:34:09 -07:00

test_webgpu.js

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

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