gitextract_4mb9jrxp/ ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── Makefile ├── README.md ├── docs/ │ └── ROADMAP.md ├── examples/ │ ├── pretrain/ │ │ ├── README.md │ │ ├── preprocess.py │ │ └── pretrain.py │ ├── usage.py │ ├── verification/ │ │ └── verify.py │ └── xla-reference/ │ ├── README.md │ ├── xpreprocess.py │ └── xpretrain.py ├── pyproject.toml ├── setup.py └── voltron/ ├── __init__.py ├── conf/ │ ├── __init__.py │ ├── accelerators.py │ ├── datasets.py │ ├── models.py │ └── tracking.py ├── datasets/ │ ├── __init__.py │ ├── datasets.py │ └── v1/ │ ├── __init__.py │ └── stream_datasets.py ├── models/ │ ├── __init__.py │ ├── core/ │ │ ├── __init__.py │ │ ├── vcond.py │ │ ├── vdual.py │ │ └── vgen.py │ ├── instantiate.py │ ├── materialize.py │ ├── reproductions/ │ │ ├── __init__.py │ │ ├── vmvp.py │ │ ├── vr3m.py │ │ └── vrn3m.py │ └── util/ │ ├── __init__.py │ ├── extraction.py │ ├── optimization.py │ └── transformer.py ├── overwatch/ │ ├── __init__.py │ └── overwatch.py ├── preprocessing/ │ ├── __init__.py │ ├── core.py │ ├── process.py │ ├── transforms.py │ └── v1/ │ ├── __init__.py │ ├── process.py │ ├── transforms.py │ └── utils.py └── util/ ├── __init__.py ├── checkpointing.py ├── metrics.py ├── utilities.py └── v1/ ├── __init__.py ├── checkpointing.py ├── distributed.py ├── random.py └── xla_logger.py