Here is a list of all namespace functions with links to the namespace documentation for each function:

- _ -

• _extract_state_dict() : export
• _write_f32_array() : export
• _write_u32s() : export

- a -

• add_loss_terms() : train
• arch_string() : train
• available() : Tablebase

- b -

• base_loss() : loss
• base_loss_components() : loss
• build_loader() : train
• build_phase14_datasets() : train
• build_scheduler() : train

- c -

• collapse_top2_if_dominant() : MoERouting
• copy_expert0_to_all() : train

- d -

• decode_feature_row() : dataset

- e -

• empty_loss_terms() : train
• export_inputs_and_refs() : export
• export_native_weights() : export

- f -

• finalize_loss_terms() : train
• format_loss_terms() : train
• format_time() : train
• forward_backbone() : train

- g -

• gater_loss() : loss
• generateCaptures() : MoveGen
• generateLegalMoves() : MoveGen
• get_device() : train

- i -

• init() : Tablebase
• init_wandb() : train
• isInitialized() : Tablebase

- k -

• killer_loss() : loss

- l -

• load_checkpoint_into_model() : train

- m -

• main() : export, prepare_dataset, train
• maxPieces() : Tablebase
• moe_loss() : loss

- p -

• parse_data_roots() : train
• phase1_export_state_dict() : train
• phase_eta_min() : train
• print_header() : train
• probeRoot() : Tablebase, TablebaseFallback
• probeWDL() : Tablebase

- r -

• resolve_data_splits() : train
• route_top2_base_experts() : MoERouting
• run_phase1() : train
• run_phase2() : train
• run_phase4() : train

- s -

• save_model_checkpoint() : train
• set_seed() : utils
• split_steps_for_epoch() : train
• survivor_loss() : loss

- t -

• to_planes_list() : train
• top2_sparse_weights() : train
• train_expert_steps() : train
• train_phase1_steps() : train
• train_phase4_steps() : train

- u -

• unpack_common_batch() : train

- v -

• validate_expert() : train
• validate_phase1() : train
• validate_phase4() : train

- w -

• wdlToScore() : Tablebase