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 datamodule_tests lightning_module_tests text_classifier text_classifier_test configs config config_test conftest datamodules image_classification cifar10 cifar10_test fashion_mnist fashion_mnist_test image_classification image_classification_test imagenet imagenet_test inaturalist inaturalist_test mnist mnist_test text text_classification text_classification_test vision vision_test 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