MANIFEST.in
pyproject.toml
setup.cfg
setup.py
/Users/matbet/Downloads/supplementary_material_hetgps/code/tensordict/tensordict/csrc/pybind.cpp
tensordict/__init__.py
tensordict/_contextlib.py
tensordict/_lazy.py
tensordict/_pytree.py
tensordict/_td.py
tensordict/_torch_func.py
tensordict/base.py
tensordict/functional.py
tensordict/memmap.py
tensordict/memmap_deprec.py
tensordict/persistent.py
tensordict/tensorclass.py
tensordict/tensordict.py
tensordict/utils.py
tensordict/version.py
tensordict.egg-info/PKG-INFO
tensordict.egg-info/SOURCES.txt
tensordict.egg-info/dependency_links.txt
tensordict.egg-info/not-zip-safe
tensordict.egg-info/requires.txt
tensordict.egg-info/top_level.txt
tensordict/nn/__init__.py
tensordict/nn/common.py
tensordict/nn/ensemble.py
tensordict/nn/functional_modules.py
tensordict/nn/params.py
tensordict/nn/probabilistic.py
tensordict/nn/sequence.py
tensordict/nn/utils.py
tensordict/nn/distributions/__init__.py
tensordict/nn/distributions/composite.py
tensordict/nn/distributions/continuous.py
tensordict/nn/distributions/discrete.py
tensordict/nn/distributions/truncated_normal.py
tensordict/nn/distributions/utils.py
tensordict/prototype/__init__.py
tensordict/prototype/fx.py
tensordict/prototype/tensorclass.py