You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman f77c04ff36 Torchscript fixes/hacks for rms_norm, refactor ParallelScalingBlock with manual combination of input projections, closer paper match		2 years ago
..
data	…
layers	Torchscript fixes/hacks for rms_norm, refactor ParallelScalingBlock with manual combination of input projections, closer paper match	2 years ago
loss	…
models	Torchscript fixes/hacks for rms_norm, refactor ParallelScalingBlock with manual combination of input projections, closer paper match	2 years ago
optim	Add multi-tensor (foreach) version of Lion in style of upcoming PyTorch 2.0 optimizers	2 years ago
scheduler	…
utils	…
__init__.py	…
version.py	Version 0.8.12dev0	2 years ago