MANIFEST.in
README.md
setup.cfg
setup.py
oslo/__init__.py
oslo/__version__.py
oslo/fused_kernels_utils.py
oslo/generation_utils.py
oslo/modeling_utils.py
oslo/parallelism_utils.py
oslo.egg-info/SOURCES.txt
oslo.egg-info/dependency_links.txt
oslo.egg-info/requires.txt
oslo.egg-info/top_level.txt
oslo/data/__init__.py
oslo/data/datasets/__init__.py
oslo/data/datasets/dataset_causal_lm.py
oslo/data/indexing/__init__.py
oslo/data/indexing/cached_indexing.py
oslo/data/indexing/lazy_indexing.py
oslo/data/indexing/mmap_indexing.py
oslo/data/preprocess/__init__.py
oslo/data/preprocess/binarizer.py
oslo/data/preprocess/encoder.py
oslo/data/preprocess/preprocessor.py
oslo/data/utils/__init__.py
oslo/data/utils/blenders.py
oslo/data/utils/loaders.py
oslo/data/utils/samplers.py
oslo/fused_kernels/__init__.py
oslo/fused_kernels/fused_ngram_repeat_block.cu
oslo/fused_kernels/fused_softmax.cu
oslo/fused_kernels/fused_triang_softmax.cu
oslo/fused_kernels/bindings/__init__.py
oslo/fused_kernels/bindings/pybind_datasets.cpp
oslo/fused_kernels/bindings/pybind_kernels.cpp
oslo/fused_kernels/includes/compat.h
oslo/fused_kernels/includes/fused_softmax.h
oslo/fused_kernels/includes/fused_triang_softmax.h
oslo/fused_kernels/includes/tqdm.h
oslo/fused_kernels/includes/type_shim.h
oslo/models/__init__.py
oslo/models/gpt2/__init__.py
oslo/models/gpt2/configuration_gpt2.py
oslo/models/gpt2/modeling_gpt2.py
oslo/models/gpt_neo/__init__.py
oslo/models/gpt_neo/configuration_gpt_neo.py
oslo/models/gpt_neo/modeling_gpt_neo.py
oslo/models/gptj/__init__.py
oslo/models/gptj/configuration_gptj.py
oslo/models/gptj/modeling_gptj.py
oslo/parallelism/__init__.py
oslo/parallelism/layer_policy.py
oslo/parallelism/mpu.py
oslo/parallelism/pipeline_parallelism.py
oslo/parallelism/tensor_parallelism.py
oslo_core.egg-info/PKG-INFO
oslo_core.egg-info/SOURCES.txt
oslo_core.egg-info/dependency_links.txt
oslo_core.egg-info/not-zip-safe
oslo_core.egg-info/requires.txt
oslo_core.egg-info/top_level.txt