Ross Wightman 5f10450235 Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now.

2025-01-27 21:02:26 -08:00

..

__init__.py

…

test_layers.py

Merge branch 'main' into fix-mqa-v2

2025-01-02 00:11:22 +01:00

test_models.py

Add num_classes assertion after reset_classifier

2025-01-21 11:52:05 -08:00

test_optim.py

Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now.

2025-01-27 21:02:26 -08:00

test_utils.py

extend existing unittests

2024-11-12 07:53:39 -08:00