Ross Wightman 1d3ab176bc Remove debug / staging code		2024-05-10 22:16:34 -07:00
..
data	…
layers	Remove JIT activations, take jit out of ME activations. Remove other instances of torch.jit.script. Breaks torch.compile and is much less performant. Remove SpaceToDepthModule	2024-05-06 16:32:49 -07:00
loss	…
models	Remove debug / staging code	2024-05-10 22:16:34 -07:00
optim	…
scheduler	…
utils	Add AttentionExtract helper module	2024-05-04 14:10:00 -07:00
__init__.py	…
version.py	Prepping weight push, benchmarking.	2024-05-10 14:14:06 -07:00