opentau 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (108) hide show
  1. opentau/__init__.py +179 -0
  2. opentau/__version__.py +24 -0
  3. opentau/configs/__init__.py +19 -0
  4. opentau/configs/default.py +297 -0
  5. opentau/configs/libero.py +113 -0
  6. opentau/configs/parser.py +393 -0
  7. opentau/configs/policies.py +297 -0
  8. opentau/configs/reward.py +42 -0
  9. opentau/configs/train.py +370 -0
  10. opentau/configs/types.py +76 -0
  11. opentau/constants.py +52 -0
  12. opentau/datasets/__init__.py +84 -0
  13. opentau/datasets/backward_compatibility.py +78 -0
  14. opentau/datasets/compute_stats.py +333 -0
  15. opentau/datasets/dataset_mixture.py +460 -0
  16. opentau/datasets/factory.py +232 -0
  17. opentau/datasets/grounding/__init__.py +67 -0
  18. opentau/datasets/grounding/base.py +154 -0
  19. opentau/datasets/grounding/clevr.py +110 -0
  20. opentau/datasets/grounding/cocoqa.py +130 -0
  21. opentau/datasets/grounding/dummy.py +101 -0
  22. opentau/datasets/grounding/pixmo.py +177 -0
  23. opentau/datasets/grounding/vsr.py +141 -0
  24. opentau/datasets/image_writer.py +304 -0
  25. opentau/datasets/lerobot_dataset.py +1910 -0
  26. opentau/datasets/online_buffer.py +442 -0
  27. opentau/datasets/push_dataset_to_hub/utils.py +132 -0
  28. opentau/datasets/sampler.py +99 -0
  29. opentau/datasets/standard_data_format_mapping.py +278 -0
  30. opentau/datasets/transforms.py +330 -0
  31. opentau/datasets/utils.py +1243 -0
  32. opentau/datasets/v2/batch_convert_dataset_v1_to_v2.py +887 -0
  33. opentau/datasets/v2/convert_dataset_v1_to_v2.py +829 -0
  34. opentau/datasets/v21/_remove_language_instruction.py +109 -0
  35. opentau/datasets/v21/batch_convert_dataset_v20_to_v21.py +60 -0
  36. opentau/datasets/v21/convert_dataset_v20_to_v21.py +183 -0
  37. opentau/datasets/v21/convert_stats.py +150 -0
  38. opentau/datasets/video_utils.py +597 -0
  39. opentau/envs/__init__.py +18 -0
  40. opentau/envs/configs.py +178 -0
  41. opentau/envs/factory.py +99 -0
  42. opentau/envs/libero.py +439 -0
  43. opentau/envs/utils.py +204 -0
  44. opentau/optim/__init__.py +16 -0
  45. opentau/optim/factory.py +43 -0
  46. opentau/optim/optimizers.py +121 -0
  47. opentau/optim/schedulers.py +140 -0
  48. opentau/planner/__init__.py +82 -0
  49. opentau/planner/high_level_planner.py +366 -0
  50. opentau/planner/utils/memory.py +64 -0
  51. opentau/planner/utils/utils.py +65 -0
  52. opentau/policies/__init__.py +24 -0
  53. opentau/policies/factory.py +172 -0
  54. opentau/policies/normalize.py +315 -0
  55. opentau/policies/pi0/__init__.py +19 -0
  56. opentau/policies/pi0/configuration_pi0.py +250 -0
  57. opentau/policies/pi0/modeling_pi0.py +994 -0
  58. opentau/policies/pi0/paligemma_with_expert.py +516 -0
  59. opentau/policies/pi05/__init__.py +20 -0
  60. opentau/policies/pi05/configuration_pi05.py +231 -0
  61. opentau/policies/pi05/modeling_pi05.py +1257 -0
  62. opentau/policies/pi05/paligemma_with_expert.py +572 -0
  63. opentau/policies/pretrained.py +315 -0
  64. opentau/policies/utils.py +123 -0
  65. opentau/policies/value/__init__.py +18 -0
  66. opentau/policies/value/configuration_value.py +170 -0
  67. opentau/policies/value/modeling_value.py +512 -0
  68. opentau/policies/value/reward.py +87 -0
  69. opentau/policies/value/siglip_gemma.py +221 -0
  70. opentau/scripts/actions_mse_loss.py +89 -0
  71. opentau/scripts/bin_to_safetensors.py +116 -0
  72. opentau/scripts/compute_max_token_length.py +111 -0
  73. opentau/scripts/display_sys_info.py +90 -0
  74. opentau/scripts/download_libero_benchmarks.py +54 -0
  75. opentau/scripts/eval.py +877 -0
  76. opentau/scripts/export_to_onnx.py +180 -0
  77. opentau/scripts/fake_tensor_training.py +87 -0
  78. opentau/scripts/get_advantage_and_percentiles.py +220 -0
  79. opentau/scripts/high_level_planner_inference.py +114 -0
  80. opentau/scripts/inference.py +70 -0
  81. opentau/scripts/launch_train.py +63 -0
  82. opentau/scripts/libero_simulation_parallel.py +356 -0
  83. opentau/scripts/libero_simulation_sequential.py +122 -0
  84. opentau/scripts/nav_high_level_planner_inference.py +61 -0
  85. opentau/scripts/train.py +379 -0
  86. opentau/scripts/visualize_dataset.py +294 -0
  87. opentau/scripts/visualize_dataset_html.py +507 -0
  88. opentau/scripts/zero_to_fp32.py +760 -0
  89. opentau/utils/__init__.py +20 -0
  90. opentau/utils/accelerate_utils.py +79 -0
  91. opentau/utils/benchmark.py +98 -0
  92. opentau/utils/fake_tensor.py +81 -0
  93. opentau/utils/hub.py +209 -0
  94. opentau/utils/import_utils.py +79 -0
  95. opentau/utils/io_utils.py +137 -0
  96. opentau/utils/libero.py +214 -0
  97. opentau/utils/libero_dataset_recorder.py +460 -0
  98. opentau/utils/logging_utils.py +180 -0
  99. opentau/utils/monkey_patch.py +278 -0
  100. opentau/utils/random_utils.py +244 -0
  101. opentau/utils/train_utils.py +198 -0
  102. opentau/utils/utils.py +471 -0
  103. opentau-0.1.0.dist-info/METADATA +161 -0
  104. opentau-0.1.0.dist-info/RECORD +108 -0
  105. opentau-0.1.0.dist-info/WHEEL +5 -0
  106. opentau-0.1.0.dist-info/entry_points.txt +2 -0
  107. opentau-0.1.0.dist-info/licenses/LICENSE +508 -0
  108. opentau-0.1.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,161 @@
1
+ Metadata-Version: 2.4
2
+ Name: opentau
3
+ Version: 0.1.0
4
+ Summary: OpenTau: Tensor's VLA Training Infrastructure for Real-World Robotics in Pytorch
5
+ Author-email: Shuheng Liu <wish1104@icloud.com>, William Yue <williamyue37@gmail.com>, Akshay Shah <akshayhitendrashah@gmail.com>, Xingrui Gu <xingrui_gu@berkeley.edu>
6
+ License: Apache-2.0
7
+ Project-URL: homepage, https://github.com/TensorAuto/OpenTau
8
+ Project-URL: issues, https://github.com/TensorAuto/OpenTau/issues
9
+ Project-URL: huggingface, https://huggingface.co/TensorAuto
10
+ Keywords: robotics,deep learning,pytorch
11
+ Classifier: Development Status :: 3 - Alpha
12
+ Classifier: Intended Audience :: Developers
13
+ Classifier: Intended Audience :: Education
14
+ Classifier: Intended Audience :: Science/Research
15
+ Classifier: Topic :: Software Development :: Build Tools
16
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
17
+ Classifier: License :: OSI Approved :: Apache Software License
18
+ Classifier: Programming Language :: Python :: 3.10
19
+ Requires-Python: ==3.10.*
20
+ Description-Content-Type: text/markdown
21
+ License-File: LICENSE
22
+ Requires-Dist: cmake>=3.29.0.1
23
+ Requires-Dist: datasets>=2.19.0
24
+ Requires-Dist: deepdiff>=7.0.1
25
+ Requires-Dist: diffusers>=0.27.2
26
+ Requires-Dist: draccus>=0.10.0
27
+ Requires-Dist: einops>=0.8.0
28
+ Requires-Dist: flask>=3.0.3
29
+ Requires-Dist: gdown>=5.1.0
30
+ Requires-Dist: h5py>=3.10.0
31
+ Requires-Dist: huggingface-hub[cli,hf-transfer]>=0.27.1; python_version < "4.0"
32
+ Requires-Dist: imageio[ffmpeg]>=2.34.0
33
+ Requires-Dist: jsonlines>=4.0.0
34
+ Requires-Dist: numba>=0.62.0
35
+ Requires-Dist: omegaconf>=2.3.0
36
+ Requires-Dist: opencv-python-headless>=4.9.0
37
+ Requires-Dist: packaging>=24.2
38
+ Requires-Dist: av>=12.0.5
39
+ Requires-Dist: pymunk>=6.6.0
40
+ Requires-Dist: pynput>=1.7.7
41
+ Requires-Dist: pyzmq>=26.2.1
42
+ Requires-Dist: rerun-sdk>=0.21.0
43
+ Requires-Dist: termcolor>=2.4.0
44
+ Requires-Dist: torch<2.8.0,>=2.7.1
45
+ Requires-Dist: torchcodec<0.5.0,>=0.4.0; sys_platform != "win32" and (sys_platform != "linux" or (platform_machine != "aarch64" and platform_machine != "arm64" and platform_machine != "armv7l")) and (sys_platform != "darwin" or platform_machine != "x86_64")
46
+ Requires-Dist: torchvision<0.23.0,>=0.22.1
47
+ Requires-Dist: wandb>=0.16.3
48
+ Requires-Dist: zarr>=2.17.0
49
+ Requires-Dist: scikit-learn>=1.7.1
50
+ Requires-Dist: onnx>=1.18.0
51
+ Requires-Dist: onnxruntime>=1.22.1; sys_platform == "darwin" or platform_machine == "arm64" or platform_machine == "aarch64"
52
+ Requires-Dist: onnxruntime-gpu>=1.22.0; (sys_platform == "linux" and platform_machine == "x86_64") or (sys_platform == "win32" and (platform_machine == "AMD64" or platform_machine == "x86_64"))
53
+ Requires-Dist: onnxscript>=0.3.1
54
+ Requires-Dist: onnx-ir>=0.1.4
55
+ Requires-Dist: opentau-transformers==4.53.3
56
+ Requires-Dist: scipy>=1.15.2
57
+ Requires-Dist: pytest>=8.1.0
58
+ Requires-Dist: pytest-cov>=5.0.0
59
+ Requires-Dist: pyserial>=3.5
60
+ Requires-Dist: pytest-xdist>=3.8.0
61
+ Requires-Dist: scikit-image>=0.23.2
62
+ Requires-Dist: pandas>=2.2.2
63
+ Requires-Dist: accelerate>=1.4.0
64
+ Requires-Dist: deepspeed>=0.17.1
65
+ Provides-Extra: dev
66
+ Requires-Dist: pre-commit>=3.7.0; extra == "dev"
67
+ Requires-Dist: debugpy>=1.8.1; extra == "dev"
68
+ Requires-Dist: pytest>=8.1.0; extra == "dev"
69
+ Requires-Dist: pytest-cov>=5.0.0; extra == "dev"
70
+ Requires-Dist: pyserial>=3.5; extra == "dev"
71
+ Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
72
+ Requires-Dist: sphinx>=8.1.3; extra == "dev"
73
+ Requires-Dist: sphinx-rtd-theme>=3.0.1; extra == "dev"
74
+ Requires-Dist: sphinx-copybutton>=0.5.2; extra == "dev"
75
+ Requires-Dist: sphinx-design>=0.6.1; extra == "dev"
76
+ Requires-Dist: myst-parser>=4.0.0; extra == "dev"
77
+ Provides-Extra: openai
78
+ Requires-Dist: openai>=1.0.0; extra == "openai"
79
+ Requires-Dist: python-dotenv>=1.0.0; extra == "openai"
80
+ Provides-Extra: libero
81
+ Requires-Dist: cmake<4; extra == "libero"
82
+ Requires-Dist: ninja; extra == "libero"
83
+ Requires-Dist: bddl==1.0.1; extra == "libero"
84
+ Requires-Dist: easydict==1.9; extra == "libero"
85
+ Requires-Dist: future==0.18.2; extra == "libero"
86
+ Requires-Dist: matplotlib>=3.5.3; extra == "libero"
87
+ Requires-Dist: robomimic==0.2.0; extra == "libero"
88
+ Requires-Dist: robosuite==1.4.0; extra == "libero"
89
+ Requires-Dist: thop==0.1.1.post2209072238; extra == "libero"
90
+ Requires-Dist: mujoco>=3.3.5; extra == "libero"
91
+ Requires-Dist: PyOpenGL==3.1.7; extra == "libero"
92
+ Requires-Dist: libero; extra == "libero"
93
+ Requires-Dist: numpy<2; extra == "libero"
94
+ Requires-Dist: gym<0.27,>=0.25; extra == "libero"
95
+ Requires-Dist: pyopengl-accelerate==3.1.7; sys_platform == "linux" and extra == "libero"
96
+ Requires-Dist: gymnasium[other]>=0.29; extra == "libero"
97
+ Dynamic: license-file
98
+
99
+ <p align="center">
100
+ <a href="https://www.tensor.auto">
101
+ <img src="assets/logo.png" alt="Logo">
102
+ </a>
103
+ </p>
104
+
105
+ # OpenTau - Train VLA models with state-of-the-art techniques by Tensor
106
+
107
+ At Tensor, we are pushing the frontier of large foundation models for physical AI. In robot learning, a vision-language-action (VLA) model is a multimodal foundation model that integrates vision, language, and action. Today, VLA represents the leading approach for embodied AI, spanning autonomous driving, robot manipulation, and navigation.
108
+
109
+ OpenTau is Tensor’s open-source training toolchain for frontier VLA models—designed to make training reproducible, accessible, and scalable. At Tensor, we believe in open research and reproducible progress for the robotics community. By open-sourcing our training toolchain, we aim to expand knowledge sharing and accelerate scientific progress that others can reproduce.
110
+
111
+ Whether you use the official OpenPi codebase or LeRobot’s reimplementation, you may still be missing key components. OpenTau implements these key capabilities in one place:
112
+
113
+ - Co-training on an adjustable mixture of heterogeneous datasets
114
+ - Discrete actions for fast VLM convergence in $\pi_{0.5}$
115
+ - Knowledge insulation between the VLM backbone and the action expert
116
+ - Dropout in the VLM to reduce overfitting
117
+ - A reinforcement learning pipeline described in $\pi^*_{0.6}$
118
+ - And more...
119
+
120
+ OpenTau ($\tau$) is a tool developed by *[Tensor][1]* to bridge this gap, and we also use it internally to train our proprietary in-house models. Our goal is to help you train VLAs on any dataset while fully leveraging state-of-the-art techniques. We plan to continuously upgrade this repository to keep pace with the state of the art in the robotics community.
121
+
122
+ | Features | OpenPi | LeRobot | **OpenTau** |
123
+ | -------------------------------------------------------: | :---------------------: | :------------------------------: | :---------: |
124
+ | Co-training with Heterogeneous Datasets | ❌ | ❌ | ✅ |
125
+ | Discrete Actions Training in $\pi_{0.5}$ | ❌ | ❌ | ✅ |
126
+ | Knowledge Insulation (KI) between VLM and Action Decoder | ❌ | ❌ | ✅ |
127
+ | Dropout Layers in PaliGemma | ✅ (Jax) <br>❌ (PyTorch) | ❌ | ✅ |
128
+ | Multi-Node and Multi-GPU Training | ❌ | ✅ | ✅ |
129
+ | Fully Functioning $\pi_{0.5}$ Checkpoint | ✅ | ❌ <br> (Missing Text Embeddings) | ✅ |
130
+ | Simulation Environments for Evaluating Models | ❌ | ✅ | ✅ |
131
+ | $\pi^{*}_{0.6}$ style Reinforcement Learning Pipeline | ❌ | ❌ | ✅ |
132
+ | Framework | Jax / PyTorch | PyTorch | PyTorch |
133
+
134
+ ## Quick Start
135
+ If you are familiar with LeRobot, getting started with OpenTau is very easy.
136
+ Because OpenTau is a fork of the popular LeRobot repository, any LeRobot-compliant policy and dataset can be used directly with OpenTau.
137
+ Check out our documentation to get started quickly.
138
+ We provide a quick start guide to help you get started with OpenTau.
139
+
140
+ For using local notebooks to train and evaluate models, find the notebooks at `notebooks/pi05_training.ipynb` and `notebooks/pi05_evaluation_only.ipynb`.
141
+
142
+ For using the Google Colab notebooks to train and evaluate models, find the colab notebooks here: [pi05_training](https://colab.research.google.com/drive/1DeU0lNnEzs1KHo0Nkgh4YKBr-xu9moBM?usp=sharing) and [pi05_evaluation_only](https://colab.research.google.com/drive/1U_AyuH9WYMT4anEWvsOtIT7g01jA0WGm?usp=sharing) respectively.
143
+
144
+ ## Checkpoints
145
+ We provide fully functioning $\pi_{0.5}$ checkpoints trained with high success rates. We plan to release more models in the near future.
146
+
147
+ | Model Checkpoint | Description | Success Rate (%) |
148
+ |-------------------------------|---------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------|
149
+ | [TensorAuto/tPi0.5-libero][2] | A $\pi_{0.5}$ model checkpoint trained on the LIBERO dataset with discrete actions and knowledge insulation. | 98.4% (10) <br> 97.6% (Goal) <br> 100% (Object) <br> 98% (Spatial) |
150
+ | [TensorAuto/pi05_base][5] | A $\pi_{0.5}$ model checkpoint converted from the official openpi checkpoint, with language embeddings added. | N/A |
151
+ | More coming soon... | | |
152
+
153
+ ## Acknowledgements
154
+
155
+ This project builds on the $\pi$ series of [papers][3] and many other open-source efforts—especially [LeRobot][4]—for re-implementing the $\pi$ models and helping standardize training infrastructure. OpenTau extends these foundations to provide a more accessible, comprehensive toolchain for training vision-language-action agents.
156
+
157
+ [1]: https://www.tensor.ai
158
+ [2]: https://huggingface.co/TensorAuto/tPi0.5-libero
159
+ [3]: https://www.pi.website/blog
160
+ [4]: https://huggingface.co/lerobot
161
+ [5]: https://huggingface.co/TensorAuto/pi05_base
@@ -0,0 +1,108 @@
1
+ opentau/__init__.py,sha256=KktmFuQQqDhhzcJKczJ99crma9qjxiJZUPv-_d0ZeLU,6697
2
+ opentau/__version__.py,sha256=junxoss59Jz_hmg3YnzhpVk_Q5Fo6uha23P1ET81N1c,889
3
+ opentau/constants.py,sha256=-_CbJujCp6hbBjJHgYMguCTcSAkVkmdpM4wHqZp7vRQ,2020
4
+ opentau/configs/__init__.py,sha256=hC-KkeCfq1mtMw9WjPCZfOTxrzQWW7hAa1w8BRC_Bqw,784
5
+ opentau/configs/default.py,sha256=MEoZyzK8olVXVHDy3FHQKK43-ULq2UQv3almymbmdCI,14387
6
+ opentau/configs/libero.py,sha256=CrRfiCBYOw7hVqv6orH_ahNyQudj6iyqHtZM9YpdvzE,4688
7
+ opentau/configs/parser.py,sha256=Pb7sw6yx38F31Aqw1J7wK6BRzfBA7DutywShyP_t9bY,14890
8
+ opentau/configs/policies.py,sha256=06oUJx0B4V6krRwyjH1goTYM3RIpRozg4SSwcVJurG4,11667
9
+ opentau/configs/reward.py,sha256=t7S8_RpEy31fAP4v_ygB-ETvaUR6OyrXmS1JNSz3cOk,1537
10
+ opentau/configs/train.py,sha256=a-c-s2zpCUkK8n0DqBPGGaJ87UcVNF02RcLa1pH843Y,18049
11
+ opentau/configs/types.py,sha256=DvKasR2v0ecSmozL0YD4S-64OeuDYhVBhtspxUDV5u0,2453
12
+ opentau/datasets/__init__.py,sha256=oLfV9vfFOg7o2XIRFiN5zOf529FafIkPwqFG7iUX4gc,4248
13
+ opentau/datasets/backward_compatibility.py,sha256=ENVQk9QDPCip6NfAxNF6Vo4VvyCWb0acV-ZxcJBsB6o,3459
14
+ opentau/datasets/compute_stats.py,sha256=N359TDuJicLKMtxxy0JVEcUtnTOB57gL5G8e9Dq0gMQ,13069
15
+ opentau/datasets/dataset_mixture.py,sha256=8UWjY9oKn9jEMe-e9Dy6no1p_21H0kXKv8A10Ku_8_o,19850
16
+ opentau/datasets/factory.py,sha256=NKWpbuNBve0PsmK1midj8g1IpQapeHn-VrxCOC3X4eI,10480
17
+ opentau/datasets/image_writer.py,sha256=JYCkImHFYpLuE88t16cYqXqQS7EHS7g6kLWXPCJmWgw,11072
18
+ opentau/datasets/lerobot_dataset.py,sha256=rz_3BcXqpcIzYr0NEVmkfLf1dY7vcTdo6zuV1CZkIuI,84747
19
+ opentau/datasets/online_buffer.py,sha256=x14P8tBz25s-hRlE8loFJs5CAvh65RGWeogF271hiF0,19671
20
+ opentau/datasets/sampler.py,sha256=5g-6prsWItVjqkt1J7mA9JPNQPhDSFx3r6rA4JphP9U,4012
21
+ opentau/datasets/standard_data_format_mapping.py,sha256=wEKilksMUjJGeIhvyLuR9qhyhtiJMK1e1AzCkbyx-l4,9667
22
+ opentau/datasets/transforms.py,sha256=pr_8vOEDUoWu7aOUdnI0_wgetsFuie3I2UYFrcStG1k,12976
23
+ opentau/datasets/utils.py,sha256=bZ0Q8KPZMWe9fLdrqJqslgDxI9sa8uxqPQTxEyWwDKw,45062
24
+ opentau/datasets/video_utils.py,sha256=NY20Et6SKWLdG4EjTNdXhpPqWEFON-UccIn_P2YukSQ,21810
25
+ opentau/datasets/grounding/__init__.py,sha256=ojvdscCIjp5EQxptFAjPgvjKGZa_Xk9LLZ2wNUebWFw,3139
26
+ opentau/datasets/grounding/base.py,sha256=FDAn2QPQHNBB7mzD7IQ2Bz882Dt1RPasBTgskXqKbP4,5773
27
+ opentau/datasets/grounding/clevr.py,sha256=lNZ0hr5EhQKTh-eg35zujybcAo5p8JQEn9dW8DJhOjI,3983
28
+ opentau/datasets/grounding/cocoqa.py,sha256=JpqVlI704tKSF-t5p-7588-wXGEWoEKrVKBxBLFn7mc,4429
29
+ opentau/datasets/grounding/dummy.py,sha256=YcO3Z7VF2tkPWYSs4U0vxypyPqgbPcFCxwpANgivLFA,3470
30
+ opentau/datasets/grounding/pixmo.py,sha256=Tm3LE8bGq93aeXqoF9W-JwJXt45_8Y86NUd9u15UJ5w,6251
31
+ opentau/datasets/grounding/vsr.py,sha256=c-wPaPVuukM8HKjGc6tF56TjxKMyZogE62VeUj-tTrs,4985
32
+ opentau/datasets/push_dataset_to_hub/utils.py,sha256=UqWfJzpg2NvSt43xYfTFmJvoFuwPfaVfMn55NXNIz6w,5070
33
+ opentau/datasets/v2/batch_convert_dataset_v1_to_v2.py,sha256=qF6Nzb0kgO6m3Ha-JnKpGLh0KOHqMb8Sv4SNDiyERic,43111
34
+ opentau/datasets/v2/convert_dataset_v1_to_v2.py,sha256=4sJw18EVC3UQaIEBzY0STKkdKgBHlT4Gy31SXKeEXhI,31654
35
+ opentau/datasets/v21/_remove_language_instruction.py,sha256=_5BWWWTAksQ9NyT7092LhkgoHItaPwOTH4MCGhZLNOk,4051
36
+ opentau/datasets/v21/batch_convert_dataset_v20_to_v21.py,sha256=wi2M5pHGZigLlfgcDKoi0B0HY5p7ue5-R2Nw-ArOs5Q,2104
37
+ opentau/datasets/v21/convert_dataset_v20_to_v21.py,sha256=_N5Gof3tCKqHd8X9mGHKWwgMi8iUQOqBXw6ysBzFdRw,5899
38
+ opentau/datasets/v21/convert_stats.py,sha256=5AollivnhmNEIzQ6EJQqk6KA09pKS_1mCfSvoObBOQA,6210
39
+ opentau/envs/__init__.py,sha256=ty4vilywul2t4vWcIHdmqli6gu5PGprGlwV9zdMwLBE,827
40
+ opentau/envs/configs.py,sha256=eIOOz7hjGrQYM_MhUca5aTr6gPFlM_NFIMjNt1GdxPo,7390
41
+ opentau/envs/factory.py,sha256=R1bh5QttjLllTn9aVaeK7C4f6Wt7vRERniQU72L6hqs,3749
42
+ opentau/envs/libero.py,sha256=UcsEXI7zVvKoy0yN0H_58r0sscyD4e8GKZt4An59Z7M,17777
43
+ opentau/envs/utils.py,sha256=4Ety8w9luWKmzyrG5wFJHfEOyQkBUvgkFCgdIu8FaHM,8014
44
+ opentau/optim/__init__.py,sha256=lry1B13y6AlLpryHytuIHAmDs8gMgOdWYduZ3JJCMUw,726
45
+ opentau/optim/factory.py,sha256=JmvkpPbMeX_VtiENCRsmJe0ekMucuV42eYAT-mggin0,1930
46
+ opentau/optim/optimizers.py,sha256=1Egb1hzAK37Ldy8-YZ8wzIgSpobdjbfhv6_eOGrUAQM,3877
47
+ opentau/optim/schedulers.py,sha256=SUIphJKqNhsylI-7-q_16E6YgiITU5DLqPKiZRUlVyY,5049
48
+ opentau/planner/__init__.py,sha256=nFKr6MR3lqJyj5EA2n8dabd-N4JZZNcp5KXJ42Yv9ls,3694
49
+ opentau/planner/high_level_planner.py,sha256=8FCWHyzP1pXbp-Dvn_txz0lWE-IywHGLlG1WMGXq9G0,12814
50
+ opentau/planner/utils/memory.py,sha256=AbyL-yK2BAhe35jngDyl01u1EugSz3snUGf8INlFm14,2288
51
+ opentau/planner/utils/utils.py,sha256=K9DHF9jPZAYMtSqltedMQrcQa8o2Zbq8tAwJG7QvzFk,2047
52
+ opentau/policies/__init__.py,sha256=KcyqKB-fbSG06g_pVjqAmymyRwgxfMJTPujGKhUY4Uw,998
53
+ opentau/policies/factory.py,sha256=fXAw6_Aij-mcdwU5SvN32MPl7fb6Z4ywSOASCDSntZw,6632
54
+ opentau/policies/normalize.py,sha256=gHOyf0RKGAOd-SbvdbHw2kGH498q5SWzDAelY0nKxJg,13846
55
+ opentau/policies/pretrained.py,sha256=JouuyJUdS_sh2inUKgcN7feUmszbXiN7CuLm2cBSluw,12836
56
+ opentau/policies/utils.py,sha256=AHvI2eRAJAEQvQUCIBgPIxj22DVV4Vx5spooqxImYDk,4367
57
+ opentau/policies/pi0/__init__.py,sha256=1JGQnTmfJLcog8ejL9JPp55JoY7FPx4dg57RWBIzqCA,817
58
+ opentau/policies/pi0/configuration_pi0.py,sha256=94EG2QlraDsPjD0zyuGwKPqToqV_ayPcUnx-JI-Bi7I,10329
59
+ opentau/policies/pi0/modeling_pi0.py,sha256=rz1S7hDOVEv12sN0ECGupddKwQVXMqdvm7G_OooWZLA,37442
60
+ opentau/policies/pi0/paligemma_with_expert.py,sha256=j9P6SL7MVP11MgyJqNnsrZAlOctWmqwDaJJAx4z9F84,20724
61
+ opentau/policies/pi05/__init__.py,sha256=VcIjZwlRW1JChRHqAK9Vz4JAIEP40RrP-W-UdyR6xk4,821
62
+ opentau/policies/pi05/configuration_pi05.py,sha256=ucgCC3BaIC6rcnotMYbElTq7ymPZh4xDGghxCseK33M,9307
63
+ opentau/policies/pi05/modeling_pi05.py,sha256=sF4OPGQWQ0eJevGDRZDPUItBkBC5Wp6WdpwsN4uqS14,50639
64
+ opentau/policies/pi05/paligemma_with_expert.py,sha256=nxBfUwBt6S4WwPDkn8LW3lHFcOib4CtoZzJEq4VKlck,23328
65
+ opentau/policies/value/__init__.py,sha256=wUP5vdpsvfnREqt4enfwaakzJ-ynX9sLYN14t2zEtpA,772
66
+ opentau/policies/value/configuration_value.py,sha256=ApjrNKHxvjNlSZ71-BPvanNAJh9GzAK9W0JCiz3mMHs,5951
67
+ opentau/policies/value/modeling_value.py,sha256=21x2EVGFlJbLasJDGTs3b5YMrrChrkuGxrYP7wLjkCY,18594
68
+ opentau/policies/value/reward.py,sha256=1ky4u5YCMTGneKs43550w8dZmBpKf4ZEMLt0cm8xpV0,3618
69
+ opentau/policies/value/siglip_gemma.py,sha256=9pyy4DvT_ewA0KnrQ9Wq_EPnK4fXPTlGp4_SyG1K9Lo,8147
70
+ opentau/scripts/actions_mse_loss.py,sha256=cc_VmN0atyOvkLlOsp4Atf9QzgCqG4_j9Z8z7yCjzSk,3216
71
+ opentau/scripts/bin_to_safetensors.py,sha256=yB-z27Rn3RVRlCWBY7-_yaDrVJ1qCgoasVvneapuui8,5288
72
+ opentau/scripts/compute_max_token_length.py,sha256=eZS6wAcTuPACk-G5B-PjGSaSJF9QIYj25GbzAf4AwNE,3976
73
+ opentau/scripts/display_sys_info.py,sha256=J174YJ3ccBknkZtjZrIS9p5pEAZ4qMbr5cEagmKTshM,2763
74
+ opentau/scripts/download_libero_benchmarks.py,sha256=c_ZggZjFzJCqvdUFliWxhmFB_3C5KR4s_SyLjEhc6Is,1848
75
+ opentau/scripts/eval.py,sha256=QpUdxLElwUZzDkZw3tJ-B2luxh2-P5UgRMNG7dlA3V0,34592
76
+ opentau/scripts/export_to_onnx.py,sha256=diyITonnGedv5Nm4eAz77BTEwoivOKYHvqYEG9Z5b08,6369
77
+ opentau/scripts/fake_tensor_training.py,sha256=y4F3CFs2jjpIJcT1wKvsrgFEebU9QFzbayhG2XT4UM0,3123
78
+ opentau/scripts/get_advantage_and_percentiles.py,sha256=JdjlADYzdS1Jc_19H6lLYMRnPlWxeckRSUQqwqb0rC4,8993
79
+ opentau/scripts/high_level_planner_inference.py,sha256=nbXr8Hp64YGeprMTpT8kvT_NgpBlI02CUlO6Mm2Js_E,3846
80
+ opentau/scripts/inference.py,sha256=_lp9YjPzarAnjiA8k2jBlIKZxza6PEHw--UyaqLPdNo,2110
81
+ opentau/scripts/launch_train.py,sha256=ThyZ0IqRfarvD3qEqa8sazSgYYL3Bh22zz6-z1JtZBs,2066
82
+ opentau/scripts/libero_simulation_parallel.py,sha256=qMee6T0EwMoAT1J2u8X4w8rsbOJYwyqD3LRAPe2Ta1g,13105
83
+ opentau/scripts/libero_simulation_sequential.py,sha256=xFSUQEuyai20QD-pYitp-UJPGE9zlaaIu4YSO0bhYKg,4775
84
+ opentau/scripts/nav_high_level_planner_inference.py,sha256=z2WHw68NWi-fJUd5TV4CrJHzxo-L7e2UliGjfOlqifM,1878
85
+ opentau/scripts/train.py,sha256=UftedMDTlTDLidV8fMPtiCu8xK9G16ivtIqaJp38su0,16866
86
+ opentau/scripts/visualize_dataset.py,sha256=_xGfAXQqhjGYMi__6L7qRH2xS5XQ2-GQRXjNw3KXMlY,10109
87
+ opentau/scripts/visualize_dataset_html.py,sha256=gEX-E5fFqBhINthf7xLMICHySvw9e3Kcf1HPRnJIyug,17979
88
+ opentau/scripts/zero_to_fp32.py,sha256=Rkl1ZczytKix9vGMg0EELzdJYFqUM1yB9p3xvSaK9k8,33272
89
+ opentau/utils/__init__.py,sha256=hIUeGPpZHf2AVf0-5C2p0BOcY0cFHCTT5yHn-SpEPwY,856
90
+ opentau/utils/accelerate_utils.py,sha256=vXnSGo1hXCUNof-oNKLMJ_SOMjpKhpZ1gx21ObSsopI,2630
91
+ opentau/utils/benchmark.py,sha256=jVli6gdBRMXAqNM3AIi43a0N_O1CLQMbKXsPK_e2y3s,3063
92
+ opentau/utils/fake_tensor.py,sha256=H8yIJJfO2vqrmWH9jXZkHqmndbd8WsO_2QTu0BZsAWY,2668
93
+ opentau/utils/hub.py,sha256=WGppA8j6P779u5eVWrkpmAfIJYoVzjLAKpeM8hQfm90,9319
94
+ opentau/utils/import_utils.py,sha256=bcfOUoW1wX6mAC-C5ESx0ABszDDIF_3PQp3uHe_gBWc,3429
95
+ opentau/utils/io_utils.py,sha256=z5S1MUF3PDOJrDmZz5FSs9FRxyiyjphLIvkUlkx4Jzs,5386
96
+ opentau/utils/libero.py,sha256=blV5ZMtxJ_ADUoLyeE_pIZCegzUcx_vOltzAWY1zXe8,7832
97
+ opentau/utils/libero_dataset_recorder.py,sha256=oafBuusXGqbMUqZrzoBrG5C441lyy8jBM3pKKwE_9rE,17024
98
+ opentau/utils/logging_utils.py,sha256=zd7ypmk7aqVposPhA7Kg-PYrstapY4MsuTklsTD4r4o,5989
99
+ opentau/utils/monkey_patch.py,sha256=cVgZ1N-NNVnlRKPA1dwO9FM4IbxR0V_Hbil6p-6knhA,9558
100
+ opentau/utils/random_utils.py,sha256=k3Ab3Y98LozGdsBzKoP8xSsFTcnaRqUzY34BsETCrrA,9102
101
+ opentau/utils/train_utils.py,sha256=0d7yvk8wlP-75pwB55gr095b_b1sWG5nlqdVxyH6_o0,6796
102
+ opentau/utils/utils.py,sha256=DrMStfjBEkw_8WVhYMnCQJNBxMeozIJ8LBSpOtMQhFM,15760
103
+ opentau-0.1.0.dist-info/licenses/LICENSE,sha256=tl3_NkxplsgU86xSvEWnDlE1UR_JsIvGo7t4hPtsIbE,27680
104
+ opentau-0.1.0.dist-info/METADATA,sha256=D89t5Nd6jD5bsHXc532C0yypVj2fOWYeThanhqj8Q_8,10456
105
+ opentau-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
106
+ opentau-0.1.0.dist-info/entry_points.txt,sha256=Q2Jf-g98RrhE7trmkvW3KrOgSBQJfdRRgUxaraF-A3c,68
107
+ opentau-0.1.0.dist-info/top_level.txt,sha256=7_yrS4x5KSeTRr2LICTCNOZmF-1_kSOFPKHvtJPL1Dw,8
108
+ opentau-0.1.0.dist-info/RECORD,,
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (80.9.0)
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
5
+
@@ -0,0 +1,2 @@
1
+ [console_scripts]
2
+ opentau-train = opentau.scripts.launch_train:main