Index _ | A | B | C | D | E | F | G | I | M | N | P | R | S | T | U | W _ __init__() (bobbin.CronTab method) (bobbin.EvalResults method) (bobbin.ImageSummary method) (bobbin.MplImageSummary method) (bobbin.MultiDirectorySummaryWriter method) (bobbin.NullSummaryWriter method) (bobbin.SampledSet method) (bobbin.ScalarSummary method) (bobbin.ThreadedSummaryWriter method) (bobbin.TrainState method) (bobbin.TrainTask method) A add() (bobbin.CronTab method) (bobbin.SampledSet method) assert_replica_integrity() (in module bobbin) B BaseTrainTask (class in bobbin) C compute_loss() (bobbin.BaseTrainTask method) create_eval_results() (bobbin.EvalTask method) CronTab (class in bobbin) D dump_pytree_json() (in module bobbin) E EvalResults (class in bobbin) EvalTask (class in bobbin) evaluate() (bobbin.EvalTask method) F finalize_eval_results() (bobbin.EvalTask method) flatten_with_paths() (in module bobbin) G gather_from_jax_processes() (in module bobbin) get_rng_dict() (bobbin.TrainTask method) I ImageSummary (class in bobbin) initialize_train_state() (bobbin.TrainTask method) is_better_than() (bobbin.EvalResults method) is_replicated_for_pmap() (bobbin.TrainState method) M make_checkpoint_saver() (bobbin.BaseTrainTask method) make_cron_action() (bobbin.EvalTask method) make_log_writer() (bobbin.BaseTrainTask method) make_training_step_fn() (bobbin.BaseTrainTask method) model (bobbin.TrainTask property) model_vars (bobbin.TrainState property) MplImageSummary (class in bobbin) MultiDirectorySummaryWriter (class in bobbin) N nested_vars_to_paths() (in module bobbin) NullSummaryWriter (class in bobbin) P parse_pytree_json() (in module bobbin) publish_train_intermediates() (in module bobbin) publish_trainer_env_info() (in module bobbin) R read_pytree_json_file() (in module bobbin) reduce() (bobbin.EvalResults method) reduce_extra_vars() (bobbin.BaseTrainTask method) replace() (bobbin.EvalResults method) (bobbin.SampledSet method) (bobbin.TrainState method) run() (bobbin.CronTab method) S SampledSet (class in bobbin) ScalarSummary (class in bobbin) schedule() (bobbin.CronTab method) summarize_shape() (in module bobbin) T ThreadedSummaryWriter (class in bobbin) to_log_message() (bobbin.EvalResults method) total_dimensionality() (in module bobbin) tpmap() (in module bobbin) TrainState (class in bobbin) TrainTask (class in bobbin) U union() (bobbin.SampledSet method) unshard() (in module bobbin) unshard_and_reduce() (bobbin.EvalResults method) W write_pytree_json_file() (in module bobbin) write_to_tensorboard() (bobbin.EvalResults method) write_trainer_log() (bobbin.BaseTrainTask method)