11 Commits

Author	SHA1	Message	Date
Ross Wightman	490d222dd8	Fix issue taking device from V before V exists	2025-01-31 12:52:47 -08:00
Lucas Nestler	e025328f96	simplify RNG	2025-01-31 17:26:14 +01:00
Lucas Nestler	6367267298	unify RNG	2025-01-31 17:23:53 +01:00
Ross Wightman	872978ccfe	Fix comment, add 'stochastic weight decay' idea because why not	2025-01-30 18:22:36 -08:00
Ross Wightman	510bbd5389	Change start/end args	2025-01-30 18:22:36 -08:00
Ross Wightman	31831f5948	Change flattening behaviour in Kron	2025-01-30 18:22:36 -08:00
Ross Wightman	b3a83b81d6	Prep Kron for merge, add detail to attributions note, README.	2025-01-27 21:02:26 -08:00
Ross Wightman	67ef6f0a92	Move opt_einsum import back out of class __init__	2025-01-27 21:02:26 -08:00
Ross Wightman	9ab5464e4d	More additions to Kron	2025-01-27 21:02:26 -08:00
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
Ross Wightman	cd21e80d03	Fiddling with Kron (PSGD)	2025-01-27 21:02:26 -08:00