Sepehr Sameni 40a518c194

use float in resample_abs_pos_embed_nhwc ...

since F.interpolate doesn't always support BFloat16

2023-07-28 16:01:42 -07:00

..

data

Replace deprecated NumPy aliases of builtin types

2023-07-03 22:24:25 +03:00

layers

use float in resample_abs_pos_embed_nhwc

2023-07-28 16:01:42 -07:00

loss

…

models

Fix ijepa vit issue with 448 model, minor formatting fixes

2023-07-26 20:46:27 -07:00

optim

import lion in __init__.py

2023-07-27 09:26:57 +08:00

scheduler

…

utils

clip gradients with update

2023-04-19 23:36:20 -07:00

__init__.py

…

version.py

Update README, version 0.9.4dev0

2023-07-27 17:07:31 -07:00