Duplicated from simulacra-research/HamiltonZero

Twobombs
/

HamiltonZero

Model card Files Files and versions

HamiltonZero / src /kfac_jax /_src

602 kB

Ctrl+K

Ctrl+K

2 contributors

History: 1 commit

Twobombs's picture

Duplicate from simulacra-research/HamiltonZero

71a70fb 25 days ago

curvature_blocks
Duplicate from simulacra-research/HamiltonZero 25 days ago
curvature_estimator
Duplicate from simulacra-research/HamiltonZero 25 days ago
utils
Duplicate from simulacra-research/HamiltonZero 25 days ago
__init__.py

615 Bytes
Duplicate from simulacra-research/HamiltonZero 25 days ago
layers_and_loss_tags.py

15.4 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago
loss_functions.py

50.7 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago
optimizer.py

87.4 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago
patches_second_moment.py

36.8 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago
tag_graph_matcher.py

72.1 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago
tracer.py

50.8 kB
Duplicate from simulacra-research/HamiltonZero 25 days ago