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 lightning_module_tests llm_finetuning llm_finetuning_test networks fcnet no_op text_classifier text_classifier_test configs config conftest datamodules datamodule_tests 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 trainers jax_trainer utils env_vars hydra_utils remote_launcher_plugin remote_launcher_plugin_test testutils typing_utils protocols utils