nimlgen 4e0d47533e beam works with var vals (#2296) ...

* beam works with var vals

* test passes now

* better comment

* linter happy

2023-11-14 13:03:19 -05:00

..

accel

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

assembly

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

datasets

add world dataset (#2045)

2023-10-11 15:54:30 -07:00

dist

feat: remove cache_id (#2236)

2023-11-08 08:09:21 -08:00

gemm

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

models

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

optimization

beam works with var vals (#2296)

2023-11-14 13:03:19 -05:00

archprobe.py

no werror in archprobe

2023-05-03 19:34:17 +00:00

augment.py

[ready] Replacing os with pathlib (#1708)

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

dump_cache.py

remove arm64, caching for cuda (#2201)

2023-11-01 18:44:00 -07:00

export_model.py

metal indirect command buffers (#2285)

2023-11-13 17:58:26 -08:00

gradcheck.py

Fix: Jacobian tests [WIP] (#1126)

2023-07-05 15:36:22 -07:00

helpers.py

refactor/ci: delete many # type: ignore (#2281)

2023-11-12 11:04:20 -08:00

hip_wrapper.py

hip multigpu training (#1878)

2023-10-24 17:35:53 -04:00

introspection.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00

lr_scheduler.py

Add pylint trailing whitespace rule (#1314)

2023-07-21 13:37:55 -04:00

onnx_ops.py

add dilation to avg_pool2d (#2270)

2023-11-13 08:47:56 -08:00

onnx.py

Onnx 1.15.0 gogogo (#2217)

2023-11-10 15:36:48 -08:00

thneed.py

simple runtime args (#2211)

2023-11-03 12:31:29 -07:00

training.py

Replace (getenv("CI", "") != "") with helpers.CI (#2213)

2023-11-03 15:20:44 -07:00

utils.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00