Reference 🤓 project algorithms callbacks callback classification_metrics samples_per_second example example_test hf_example hf_example_test jax_example jax_example_test jax_rl_example jax_rl_example_test no_op testsuites algorithm_tests 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 hf_text hf_text_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