MANIFEST.in
README.md
pyproject.toml
requirements.txt
setup.py
requirements/core_requirements.txt
requirements/dev_requirements.txt
requirements/docs_requirements.txt
requirements/extra_requirements.txt
requirements/lint_requirements.txt
requirements/llm_eval.txt
requirements/mauve_text.txt
requirements/molgen_requirements.txt
requirements/plotting_requirements.txt
requirements/safe_extra.txt
requirements/test_requirements.txt
src/xlm/__init__.py
src/xlm/__main__.py
src/xlm/datamodule.py
src/xlm/external_models.py
src/xlm/flags.py
src/xlm/generative_perplexity.py
src/xlm/harness.py
src/xlm/log_predictions.py
src/xlm/metrics.py
src/xlm/model.py
src/xlm/noise.py
src/xlm/py.typed
src/xlm/version.py
src/xlm/backbones/__init__.py
src/xlm/backbones/dream/__init__.py
src/xlm/backbones/dream/configuration_base.py
src/xlm/backbones/dream/modeling_dream.py
src/xlm/backbones/dream/tokenization_dream.py
src/xlm/commands/__init__.py
src/xlm/commands/cli_demo.py
src/xlm/commands/extract_checkpoint.py
src/xlm/commands/extract_model_state_dict.py
src/xlm/commands/generate_star_graphs.py
src/xlm/commands/hydra_callbacks.py
src/xlm/commands/lightning_eval.py
src/xlm/commands/lightning_generate.py
src/xlm/commands/lightning_main.py
src/xlm/commands/lightning_prepare_data.py
src/xlm/commands/lightning_train.py
src/xlm/commands/push_to_hub.py
src/xlm/commands/scaffold_model.py
src/xlm/commands/split_owt.py
src/xlm/commands/train_tokenizer.py
src/xlm/configs/common/cluster/unity_default.yaml
src/xlm/configs/common/cluster/unity_gypsum.yaml
src/xlm/configs/common/hydra/default.yaml
src/xlm/configs/common/hydra/callbacks/dynamic_help.yaml
src/xlm/configs/common/hydra/help/xlm.yaml
src/xlm/configs/common/paths/default.yaml
src/xlm/configs/lightning_train/config.yaml
src/xlm/configs/lightning_train/callbacks/checkpoint_every_n_steps_with_thinning.yaml
src/xlm/configs/lightning_train/callbacks/checkpoint_monitor.yaml
src/xlm/configs/lightning_train/callbacks/debug.yaml
src/xlm/configs/lightning_train/callbacks/debug_small_data.yaml
src/xlm/configs/lightning_train/callbacks/debug_v2.yaml
src/xlm/configs/lightning_train/callbacks/default.yaml
src/xlm/configs/lightning_train/callbacks/default_hf.yaml
src/xlm/configs/lightning_train/callbacks/default_v2.yaml
src/xlm/configs/lightning_train/callbacks/ema.yaml
src/xlm/configs/lightning_train/callbacks/file_friendly_progress_bar.yaml
src/xlm/configs/lightning_train/callbacks/fsdp_diagnostics.yaml
src/xlm/configs/lightning_train/callbacks/learning_rate_monitor.yaml
src/xlm/configs/lightning_train/callbacks/log_gradients_to_tensorboard.yaml
src/xlm/configs/lightning_train/callbacks/on_exception_checkpoint.yaml
src/xlm/configs/lightning_train/callbacks/print_batch.yaml
src/xlm/configs/lightning_train/callbacks/print_batch_v2.yaml
src/xlm/configs/lightning_train/callbacks/speed_monitor.yaml
src/xlm/configs/lightning_train/callbacks/unconditional_samples.yaml
src/xlm/configs/lightning_train/datamodule/cfg.yaml
src/xlm/configs/lightning_train/datamodule/default.yaml
src/xlm/configs/lightning_train/datamodule/lm1b.yaml
src/xlm/configs/lightning_train/datamodule/owt.yaml
src/xlm/configs/lightning_train/datamodule/owt_packed.yaml
src/xlm/configs/lightning_train/datamodule/qm9.yaml
src/xlm/configs/lightning_train/datamodule/safe.yaml
src/xlm/configs/lightning_train/datamodule/safe_fragment.yaml
src/xlm/configs/lightning_train/datamodule/uniref50.yaml
src/xlm/configs/lightning_train/datamodule/uniref50_packed.yaml
src/xlm/configs/lightning_train/datasets/cfg.yaml
src/xlm/configs/lightning_train/datasets/cfg_test.yaml
src/xlm/configs/lightning_train/datasets/cfg_train.yaml
src/xlm/configs/lightning_train/datasets/cfg_val.yaml
src/xlm/configs/lightning_train/datasets/default_eval.yaml
src/xlm/configs/lightning_train/datasets/default_text.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_test.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_test_10v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_test_8v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_train.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_train_10v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_train_8v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val_10v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val_8v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val_pred.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val_pred_10v.yaml
src/xlm/configs/lightning_train/datasets/gram_graph_coloring_val_pred_8v.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_test.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_test_10x10.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_test_8x8.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_train.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_train_10x10.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_train_8x8.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val_10x10.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val_8x8.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val_pred.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val_pred_10x10.yaml
src/xlm/configs/lightning_train/datasets/gram_n_queens_val_pred_8x8.yaml
src/xlm/configs/lightning_train/datasets/lm1b_test.yaml
src/xlm/configs/lightning_train/datasets/lm1b_train.yaml
src/xlm/configs/lightning_train/datasets/math500.yaml
src/xlm/configs/lightning_train/datasets/math500_test.yaml
src/xlm/configs/lightning_train/datasets/owt_packed_test.yaml
src/xlm/configs/lightning_train/datasets/owt_packed_train.yaml
src/xlm/configs/lightning_train/datasets/owt_packed_val.yaml
src/xlm/configs/lightning_train/datasets/owt_test.yaml
src/xlm/configs/lightning_train/datasets/owt_train.yaml
src/xlm/configs/lightning_train/datasets/owt_val.yaml
src/xlm/configs/lightning_train/datasets/qm9_train.yaml
src/xlm/configs/lightning_train/datasets/qm9_val.yaml
src/xlm/configs/lightning_train/datasets/safe.yaml
src/xlm/configs/lightning_train/datasets/safe_fragment_test.yaml
src/xlm/configs/lightning_train/datasets/safe_fragment_val.yaml
src/xlm/configs/lightning_train/datasets/safe_test.yaml
src/xlm/configs/lightning_train/datasets/safe_train.yaml
src/xlm/configs/lightning_train/datasets/safe_val.yaml
src/xlm/configs/lightning_train/datasets/star.yaml
src/xlm/configs/lightning_train/datasets/star_easy_test.yaml
src/xlm/configs/lightning_train/datasets/star_easy_test_pred.yaml
src/xlm/configs/lightning_train/datasets/star_easy_train.yaml
src/xlm/configs/lightning_train/datasets/star_easy_val.yaml
src/xlm/configs/lightning_train/datasets/star_easy_val_pred.yaml
src/xlm/configs/lightning_train/datasets/star_hard_test.yaml
src/xlm/configs/lightning_train/datasets/star_hard_test_pred.yaml
src/xlm/configs/lightning_train/datasets/star_hard_train.yaml
src/xlm/configs/lightning_train/datasets/star_hard_val.yaml
src/xlm/configs/lightning_train/datasets/star_hard_val_pred.yaml
src/xlm/configs/lightning_train/datasets/star_medium_test.yaml
src/xlm/configs/lightning_train/datasets/star_medium_test_pred.yaml
src/xlm/configs/lightning_train/datasets/star_medium_train.yaml
src/xlm/configs/lightning_train/datasets/star_medium_val.yaml
src/xlm/configs/lightning_train/datasets/star_medium_val_pred.yaml
src/xlm/configs/lightning_train/datasets/sudoku_extreme_test.yaml
src/xlm/configs/lightning_train/datasets/sudoku_extreme_train.yaml
src/xlm/configs/lightning_train/datasets/sudoku_extreme_val.yaml
src/xlm/configs/lightning_train/datasets/sudoku_extreme_val_pred.yaml
src/xlm/configs/lightning_train/datasets/sudoku_kaggle_test.yaml
src/xlm/configs/lightning_train/datasets/sudoku_kaggle_train.yaml
src/xlm/configs/lightning_train/datasets/sudoku_kaggle_val.yaml
src/xlm/configs/lightning_train/datasets/sudoku_kaggle_val_pred.yaml
src/xlm/configs/lightning_train/datasets/sudoku_train.yaml
src/xlm/configs/lightning_train/datasets/sudoku_val.yaml
src/xlm/configs/lightning_train/datasets/sudoku_val_pred.yaml
src/xlm/configs/lightning_train/datasets/text_unconditional_prediction.yaml
src/xlm/configs/lightning_train/datasets/uniref50_packed_train.yaml
src/xlm/configs/lightning_train/datasets/uniref50_packed_val.yaml
src/xlm/configs/lightning_train/datasets/uniref50_train.yaml
src/xlm/configs/lightning_train/datasets/uniref50_val.yaml
src/xlm/configs/lightning_train/debug/eval_unconditional_preds.yaml
src/xlm/configs/lightning_train/debug/multinode.yaml
src/xlm/configs/lightning_train/debug/no_dl_workers.yaml
src/xlm/configs/lightning_train/debug/overfit.yaml
src/xlm/configs/lightning_train/debug/print_predictions.yaml
src/xlm/configs/lightning_train/debug/profile_single_node.yaml
src/xlm/configs/lightning_train/debug/small_data.yaml
src/xlm/configs/lightning_train/experiment/gpt2_generative_perplexity.yaml
src/xlm/configs/lightning_train/loggers/debug.yaml
src/xlm/configs/lightning_train/loggers/default.yaml
src/xlm/configs/lightning_train/loggers/tensorboard.yaml
src/xlm/configs/lightning_train/loggers/wandb.yaml
src/xlm/configs/lightning_train/metrics/accumulated_loss.yaml
src/xlm/configs/lightning_train/metrics/exact_match.yaml
src/xlm/configs/lightning_train/metrics/parsability.yaml
src/xlm/configs/lightning_train/metrics/seq2seq_exact_match.yaml
src/xlm/configs/lightning_train/metrics/seq2seq_length_loss.yaml
src/xlm/configs/lightning_train/metrics/seq2seq_token_accuracy.yaml
src/xlm/configs/lightning_train/metrics/seq2seq_token_ce.yaml
src/xlm/configs/lightning_train/metrics/steps_taken.yaml
src/xlm/configs/lightning_train/metrics/token_accuracy.yaml
src/xlm/configs/lightning_train/model/rotary_transformer.yaml
src/xlm/configs/lightning_train/model/rotary_transformer_xtiny2.yaml
src/xlm/configs/lightning_train/noise_schedule/dummy.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/default.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/denovo.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/gen_ppl_gpt2_large.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/gen_ppl_gpt2_small.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/gen_ppl_llama3_3b.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/mauve_text.yaml
src/xlm/configs/lightning_train/post_hoc_evaluator/syntactic.yaml
src/xlm/configs/lightning_train/tokenizer/cfg_small.yaml
src/xlm/configs/lightning_train/tokenizer/safe.yaml
src/xlm/configs/lightning_train/trainer_strategy/cpu.yaml
src/xlm/configs/lightning_train/trainer_strategy/ddp.yaml
src/xlm/configs/lightning_train/trainer_strategy/ddp_multinode.yaml
src/xlm/configs/lightning_train/trainer_strategy/ddp_multinode_no_hooks.yaml
src/xlm/configs/lightning_train/trainer_strategy/fsdp.yaml
src/xlm/configs/lightning_train/trainer_strategy/single_device.yaml
src/xlm/configs/slurm/generate_sbatch.yaml
src/xlm/configs/slurm/nll_sbatch.yaml
src/xlm/configs/slurm/train_sbatch.yaml
src/xlm/configs/slurm/hardware/1_node_1_gpu.yaml
src/xlm/configs/slurm/hardware/ddp_1_node_2_gpu.yaml
src/xlm/configs/slurm/hardware/ddp_1_node_8_gpu.yaml
src/xlm/configs/slurm/hardware/ddp_2_node_1_gpu_debug.yaml
src/xlm/configs/slurm/hardware/ddp_4_node_1_gpu.yaml
src/xlm/configs/slurm/hardware/ddp_8_node_1_gpu.yaml
src/xlm/modules/__init__.py
src/xlm/modules/ddit_simple.py
src/xlm/modules/ddit_simple_v2.py
src/xlm/modules/encoder.py
src/xlm/modules/gpt2_transformer.py
src/xlm/modules/position.py
src/xlm/modules/regression.py
src/xlm/modules/rotary_transformer.py
src/xlm/tasks/__init__.py
src/xlm/tasks/cfg/__init__.py
src/xlm/tasks/composite_eval/__init__.py
src/xlm/tasks/gram_graph_coloring/__init__.py
src/xlm/tasks/gram_n_queens/__init__.py
src/xlm/tasks/lm1b/__init__.py
src/xlm/tasks/math500/__init__.py
src/xlm/tasks/owt/__init__.py
src/xlm/tasks/owt/generative_perplexity_post_hoc.py
src/xlm/tasks/owt/mauve_text_eval.py
src/xlm/tasks/qm9/__init__.py
src/xlm/tasks/safe_molgen/__init__.py
src/xlm/tasks/safe_molgen/_safe_molgen_impl.py
src/xlm/tasks/safe_molgen/zinc_len.pkl
src/xlm/tasks/star/__init__.py
src/xlm/tasks/sudoku/__init__.py
src/xlm/tasks/sudoku_extreme/__init__.py
src/xlm/tasks/uniref50/__init__.py
src/xlm/utils/__init__.py
src/xlm/utils/batch_size.py
src/xlm/utils/checkpoint_paths.py
src/xlm/utils/checkpoint_with_thinning.py
src/xlm/utils/consolidate_model_checkpoint.py
src/xlm/utils/data.py
src/xlm/utils/ddp.py
src/xlm/utils/debug.py
src/xlm/utils/ema.py
src/xlm/utils/file_friendly_progress_callback.py
src/xlm/utils/flops_estimation.py
src/xlm/utils/fsdp_diagnostics_callback.py
src/xlm/utils/fsdp_grouping.py
src/xlm/utils/hf_hub.py
src/xlm/utils/imports.py
src/xlm/utils/model_loading.py
src/xlm/utils/model_state_dict.py
src/xlm/utils/nn.py
src/xlm/utils/omegaconf_resolvers.py
src/xlm/utils/on_exception_checkpoint.py
src/xlm/utils/os.py
src/xlm/utils/rank_zero.py
src/xlm/utils/rich_utils.py
src/xlm/utils/saving_utils.py
src/xlm/utils/seed.py
src/xlm/utils/signal.py
src/xlm/utils/slurm.py
src/xlm/utils/speed_monitor_callback.py
src/xlm/utils/tensorboard_logging.py
src/xlm/utils/text.py
src/xlm/utils/tqdm.py
src/xlm/utils/wandb.py
src/xlm_core.egg-info/PKG-INFO
src/xlm_core.egg-info/SOURCES.txt
src/xlm_core.egg-info/dependency_links.txt
src/xlm_core.egg-info/entry_points.txt
src/xlm_core.egg-info/requires.txt
src/xlm_core.egg-info/top_level.txt
tests/test_fsdp_grouping.py
tests/test_github_release.py