Add lion algorithm #353

Job	Run time
build_and_push_jax_docker_image	21m 57s
build_and_push_pytorch_docker_image	6m 3s
imagenet_resnet_pytorch	0s
ogbg_pytorch	0s
fastmri_pytorch	0s
criteo_pytorch	0s
librispeech_deepspeech_pytorch	0s
imagenet_vit_pytorch	0s
librispeech_conformer_pytorch	0s
wmt_pytorch	0s
fastmri_jax	1s
librispeech_conformer_jax	1s
imagenet_resnet_jax	1s
imagenet_vit_jax	1d 0h 0m 0s
ogbg_jax	1s
librispeech_deepspeech_jax	1s
criteo_jax	1d 0h 0m 0s
wmt_jax	1d 0h 0m 1s
	3d 0h 28m 6s

Navigation Menu