LICENSE
README.md
pyproject.toml
setup.py
mlmisc/__init__.py
mlmisc/args_parallel.py
mlmisc/args_sequential.py
mlmisc/attention.py
mlmisc/auto_module.py
mlmisc/config.py
mlmisc/conv2d_mixer.py
mlmisc/conv_linear.py
mlmisc/conv_utils.py
mlmisc/core_utils.py
mlmisc/data_loader.py
mlmisc/dataset_adapters.py
mlmisc/dataset_base.py
mlmisc/dataset_utils.py
mlmisc/debug_utils.py
mlmisc/einops_layers.py
mlmisc/encoder_block.py
mlmisc/ensemble.py
mlmisc/fork_attention.py
mlmisc/image_pad_concat.py
mlmisc/image_urls_dataset.py
mlmisc/layer_utils.py
mlmisc/linear_conv.py
mlmisc/load_state_dict.py
mlmisc/loss_wrappers.py
mlmisc/model_wrappers.py
mlmisc/module_builder.py
mlmisc/moe.py
mlmisc/mosaic.py
mlmisc/net_base.py
mlmisc/nets_dict.py
mlmisc/nn_lambda.py
mlmisc/patcher.py
mlmisc/sequence_datasets.py
mlmisc/sequence_projection.py
mlmisc/sequence_utils.py
mlmisc/shard_attention.py
mlmisc/split_linear.py
mlmisc/text_dataset.py
mlmisc/tiled_linear.py
mlmisc/tokenizers.py
mlmisc/torch_profiler.py
mlmisc/trainer.py
mlmisc/types.py
mlmisc/unet.py
mlmisc/utils.py
mlmisc/web_dataset.py
mlmisc/web_module.py
mlmisc.egg-info/PKG-INFO
mlmisc.egg-info/SOURCES.txt
mlmisc.egg-info/dependency_links.txt
mlmisc.egg-info/requires.txt
mlmisc.egg-info/top_level.txt
mlmisc/bin/__init__.py
mlmisc/bin/base_setup.py
mlmisc/bin/checkpoint_rewrite.py
mlmisc/bin/dataset.py
mlmisc/bin/extract_metrics.py
mlmisc/bin/extract_parameters.py
mlmisc/bin/img2img_model_test.py
mlmisc/bin/img_model_test.py
mlmisc/bin/lrsched_plot.py
mlmisc/bin/plot_setup.py
mlmisc/bin/textgen.py
mlmisc/bin/train_runner.py
mlmisc/bin/weight_rename.py
mlmisc/lrsched/__init__.py
mlmisc/lrsched/reduce_on_plateau.py
mlmisc/lrsched/wrapper.py
mlmisc/models/__init__.py
mlmisc/models/sequence/__init__.py
mlmisc/models/sequence/embedding_trainer.py
mlmisc/models/sequence/encoder_stack.py
mlmisc/models/sequence/sequence_base.py
mlmisc/models/vision/__init__.py
mlmisc/models/vision/cnnensemble.py
mlmisc/models/vision/conv_vit.py
mlmisc/models/vision/covit.py
mlmisc/models/vision/cross_viz_seq.py
mlmisc/models/vision/encdec.py
mlmisc/models/vision/fork_viz_seq.py
mlmisc/models/vision/patcher_vit.py
mlmisc/models/vision/vit_base.py
mlmisc/models/vision/vito.py
mlmisc/rl/__init__.py
mlmisc/rl/actions.py
mlmisc/rl/env.py
mlmisc/rl/main_utils.py
mlmisc/rl/memory.py
mlmisc/rl/nets.py
mlmisc/rl/param_updater.py
mlmisc/rl/train_context.py
mlmisc/rl/train_runner.py
mlmisc/rl/utils.py