Ross Wightman 71101ebba0 Refactor vit pooling to add more reduction options, separately callable		2024-06-14 23:16:58 -07:00
..
data	…
layers	Fix rotary embed version of attn pool. Bit of cleanup/naming	2024-06-11 23:49:17 -07:00
loss	…
models	Refactor vit pooling to add more reduction options, separately callable	2024-06-14 23:16:58 -07:00
optim	…
scheduler	…
utils	…
__init__.py	…
version.py	Update README.md, bump dev version 1.0.6	2024-06-12 12:35:44 -07:00