Reference 🤓 project algorithms callbacks classification_metrics samples_per_second image_classifier image_classifier_test jax_image_classifier jax_image_classifier_test jax_ppo jax_ppo_test llm_finetuning llm_finetuning_test no_op testsuites lightning_module_tests text_classifier text_classifier_test configs algorithm lr_scheduler network optimizer config config_test datamodule conftest datamodules datamodules_test image_classification cifar10 fashion_mnist image_classification imagenet inaturalist inaturalist_test mnist text text_classification text_classification_test vision experiment main main_test networks fcnet trainers jax_trainer utils env_vars hydra_utils remote_launcher_plugin remote_launcher_plugin_test testutils typing_utils jax_typing_utils protocols utils