Reference 🤓 project algorithms callbacks callback classification_metrics samples_per_second example example_test jax_example jax_example_test jax_rl_example jax_rl_example_test llm_finetuning llm_finetuning_test no_op testsuites algorithm_tests text_classification_example text_classification_example_test configs algorithm lr_scheduler network optimizer config config_test datamodule conftest datamodules datamodules_test image_classification cifar10 fashion_mnist image_classification imagenet imagenet32 imagenet32_test inaturalist inaturalist_test mnist text text_classification text_classification_test vision experiment main main_test networks fcnet trainers jax_trainer utils autoref_plugin autoref_plugin_test env_vars hydra_config_utils hydra_config_utils_test hydra_utils remote_launcher_plugin remote_launcher_plugin_test seeding testutils typing_utils jax_typing_utils protocols utils