LICENSE
README.md
pyproject.toml
socket_lm.egg-info/PKG-INFO
socket_lm.egg-info/SOURCES.txt
socket_lm.egg-info/dependency_links.txt
socket_lm.egg-info/requires.txt
socket_lm.egg-info/top_level.txt
socketimport/__init__.py
socketimport/config.py
socketimport/deps.py
socketimport/device.py
socketimport/errors.py
socketimport/utils.py
socketimport/core/__init__.py
socketimport/core/accelerator.py
socketimport/core/attention.py
socketimport/core/base.py
socketimport/core/galore.py
socketimport/core/lite_ladder.py
socketimport/core/lora.py
socketimport/core/model_loader.py
socketimport/core/quantization.py
socketimport/core/reft.py
socketimport/data/__init__.py
socketimport/data/loader.py
socketimport/distributed/__init__.py
socketimport/memory/__init__.py
socketimport/memory/batch_finder.py
socketimport/memory/checkpointing.py
socketimport/memory/tracker.py
socketimport/optim/__init__.py
socketimport/optim/ema.py
socketimport/optim/fused.py
socketimport/optim/lion.py
socketimport/optim/scheduler.py
socketimport/optim/swa.py
socketimport/profiler/__init__.py
socketimport/profiler/benchmark.py
socketimport/profiler/trace.py
socketimport/training/__init__.py
socketimport/training/callbacks.py
socketimport/training/framework_adapter.py
socketimport/training/oom_recovery.py
socketimport/training/trainer.py
socketimport/training/vram_guard.py
tests/test_accelerator.py
tests/test_attention.py
tests/test_batch_finder.py
tests/test_benchmark.py
tests/test_callbacks.py
tests/test_checkpointing.py
tests/test_config.py
tests/test_deps.py
tests/test_device.py
tests/test_distributed.py
tests/test_ema.py
tests/test_errors.py
tests/test_framework_adapter.py
tests/test_fused.py
tests/test_galore.py
tests/test_lion.py
tests/test_lite_ladder.py
tests/test_loader.py
tests/test_lora.py
tests/test_model_loader.py
tests/test_oom_recovery.py
tests/test_quantization.py
tests/test_reft.py
tests/test_scheduler.py
tests/test_swa.py
tests/test_trace.py
tests/test_tracker.py
tests/test_trainer.py
tests/test_utils.py
tests/test_vram_guard.py