Jared Kaplan db55ef4fa7 Added a Softmax Xent Op (#53) ...

Also includes a bugfix in kernel.py to set the device before registering the c++ function object

2021-02-07 12:53:42 -08:00

..

test_blocksparse.py

[PYTHON][OPS][BLOCKSPARSE] Now rounding softmax tile sizes to next power

2021-02-04 16:13:45 -05:00

test_conv.py

[PYTHON][TESTS][DOC] Various improvement of the API and code quality:

2021-01-29 17:27:16 -05:00

test_cross_entropy.py

Added a Softmax Xent Op (#53)

2021-02-07 12:53:42 -08:00

test_matmul.py

[TESTS] test_matmul.py now plots benchmarks

2021-02-04 15:35:53 -05:00

test_softmax.py

[PYTHON][TESTS][DOC] Various improvement of the API and code quality:

2021-01-29 17:27:16 -05:00