Here is a list of all namespace members with links to the namespace documentation for each member:

- _ -

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

- a -

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

- b -

• B : loss
• backbone_params : model
• base_loss() : loss
• base_loss_components() : loss
• branches_params : model
• build_loader() : train
• build_phase14_datasets() : train
• build_scheduler() : train
• bypass : model

- c -

• C : train
• CAPTURE_BASE : MoveOrderConstants
• collapse_top2_if_dominant() : MoERouting
• CONTEMPT_FACTOR : SearchConstants
• copy_expert0_to_all() : train

- d -

• decode_feature_row() : dataset
• default : dataset
• DEFAULT_DEPTH : SearchConstants
• DRAW_SCORE : SearchConstants
• ds : dataset

- e -

• empty_loss_terms() : train
• expert_params : model
• 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 -

• g_loss : loss
• gater_loss() : loss
• generateCaptures() : MoveGen
• generateLegalMoves() : MoveGen
• get_device() : train
• global_v : model

- h -

• HASH_MOVE_SCORE : MoveOrderConstants
• help : dataset
• HISTORY_MAX : MoveOrderConstants

- i -

• INFINITE : SearchConstants
• init() : Tablebase
• init_wandb() : train
• initialized_ : Tablebase
• int : dataset
• INVALID_EVAL : SearchConstants
• isInitialized() : Tablebase

- k -

• killer_loss() : loss
• KILLER_SCORE_1 : MoveOrderConstants
• KILLER_SCORE_2 : MoveOrderConstants

- l -

• LMR_MIN_DEPTH : SearchConstants
• load_checkpoint_into_model() : train
• loss : loss

- m -

• MAGIC_INPUTS : export
• MAGIC_REFS : export
• MAGIC_WEIGHTS : export
• main() : export, prepare_dataset, train
• MATE_IN_MAX : SearchConstants
• MATE_SCORE : SearchConstants
• MAX_BRANCH_PLANES : export
• MAX_EXTENSIONS : SearchConstants
• MAX_PLY : SearchConstants
• maxPieces() : Tablebase
• maxPieces_ : Tablebase
• model : model
• moe_loss() : loss

- n -

• NULL_MOVE_MIN_DEPTH : SearchConstants

- p -

• PACKED_OFFSETS_CONST : dataset
• parse_data_roots() : train
• parser : dataset
• phase1_export_state_dict() : train
• phase_eta_min() : train
• planes_list : model
• PLANES_PER_GROUP_CONST : dataset
• pointwise_mixer_params : model
• POOL_TO_CODE : export
• pred_gates : loss
• pred_wdl : loss
• print_header() : train
• probeRoot() : Tablebase, TablebaseFallback
• probeWDL() : Tablebase

- r -

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

- s -

• sample : dataset
• save_model_checkpoint() : train
• set_seed() : utils
• split_steps_for_epoch() : train
• stem_global_params : model
• survivor_loss() : loss

- t -

• target_gates : loss
• target_wdl : loss
• TB_WIN_IN_MAX : SearchConstants
• TB_WIN_SCORE : SearchConstants
• to_planes_list() : train
• top2_sparse_weights() : train
• total_params : model
• train_expert_steps() : train
• train_phase1_steps() : train
• train_phase4_steps() : train
• TT_ALPHA : SearchConstants
• TT_BETA : SearchConstants
• TT_EXACT : SearchConstants
• type : dataset

- u -

• unpack_common_batch() : train

- v -

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

- w -

• WANDB_AVAILABLE : train
• WDL : Tablebase
• wdl : model
• wdlToScore() : Tablebase