George Hotz c5edb3c374 train value net, improve API, add BCE (#2047) ...

* api cleanups, BCE losses

* valuenet

* fixup examples

* learning okay

* add valuenet runner

* net improvements

* net improvements

* 40% win rate

2023-10-12 07:56:38 -07:00

..

external

Torch eq fix (#1562)

2023-10-11 12:57:11 -07:00

extra

Allow multi-input model export (#1995)

2023-10-07 04:13:34 -07:00

models

Torch eq fix (#1562)

2023-10-11 12:57:11 -07:00

unit

Refactor and add more unit tests for disktensors (#2022)

2023-10-09 18:46:29 -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

no import when Python is shutting down (#1875)

2023-09-20 12:47:02 -04:00

test_assign.py

corealize + remove realize from lazybuffer (#1968)

2023-10-04 10:59:31 -07:00

test_cache_collector.py

fix jitted dist (#1955)

2023-10-02 11:45:13 -04:00

test_conv_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

3x3 winograd convs (#1675)

2023-09-03 07:29:43 -07:00

test_custom_function.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_dtype.py

Fix cast error in render_load in wgsl (#1956)

2023-10-04 02:29:14 -07:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

unrealized consts everywhere (#1963)

2023-10-04 01:48:10 -07:00

test_lazybuffer.py

corealize + remove realize from lazybuffer (#1968)

2023-10-04 10:59:31 -07:00

test_linearizer_failures.py

start work on auto opt (#2034)

2023-10-11 12:54:53 -07:00

test_linearizer.py

test: fix test_linearizer.test_tensor_core test (#2036)

2023-10-10 14:48:28 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

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

test_nn.py

3x3 winograd convs (#1675)

2023-09-03 07:29:43 -07:00

test_ops.py

train value net, improve API, add BCE (#2047)

2023-10-12 07:56:38 -07:00

test_optim.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

test_randomness.py

Tensor.uniform with dtype=int bug fix (#1593)

2023-08-26 01:59:53 -04:00

test_schedule.py

move image+kopt to features (#2015)

2023-10-07 15:41:08 -07:00

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

wmma: add test and tensor core shape (#1925)

2023-09-28 18:04:28 -07:00

test_symbolic_jit.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_symbolic_ops.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_symbolic_shapetracker.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_tensor.py

fix transfer to mapped buffers (#1923)

2023-09-29 00:50:24 -07:00

test_uops.py

Make Triton work again (#1547)

2023-09-23 14:17:12 +08:00

test_webgpu.js

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

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

test_winograd.py

Schedule item (#2012)

2023-10-07 08:59:25 -07:00