Update requirements.txt

requirements.txt

@@ -1,825 +1,65 @@
-# This file was autogenerated by uv via the following command:
-#    uv export --format requirements-txt --no-hashes
-    # via openpi
-absl-py==2.3.0
-    # via
-    #   chex
-    #   dm-control
-    #   dm-env
-    #   dm-tree
-    #   labmaze
-    #   ml-collections
-    #   mujoco
-    #   optax
-    #   orbax-checkpoint
-aiohappyeyeballs==2.6.1
-    # via aiohttp
-aiohttp==3.12.4
-    # via
-    #   fsspec
-    #   gcsfs
-aiosignal==1.3.2
-    # via aiohttp
-annotated-types==0.7.0
-    # via pydantic
-antlr4-python3-runtime==4.9.3
-    # via omegaconf
-appnope==0.1.4 ; sys_platform == 'darwin'
-    # via ipykernel
-asttokens==3.0.0
-    # via stack-data
-attrs==25.3.0
-    # via
-    #   aiohttp
-    #   dm-tree
-    #   jsonlines
-    #   rerun-sdk
-augmax==0.4.1
-    # via openpi
-av==14.4.0
-    # via lerobot
-beartype==0.19.0
-    # via openpi
-beautifulsoup4==4.13.4
-    # via gdown
-cachetools==5.5.2
-    # via google-auth
-certifi==2025.4.26
-    # via
-    #   requests
-    #   sentry-sdk
-cffi==1.17.1
-    # via
-    #   pymunk
-    #   pyzmq
-cfgv==3.4.0
-    # via pre-commit
-charset-normalizer==3.4.2
-    # via requests
-chex==0.1.89
-    # via optax
-click==8.2.1
-    # via
-    #   flask
-    #   tree
-    #   wandb
-cloudpickle==3.1.1
-    # via gymnasium
-cmake==4.0.2
-    # via lerobot
-colorama==0.4.6 ; sys_platform == 'win32'
-    # via
-    #   click
-    #   ipython
-    #   pytest
-    #   tqdm
-    #   tyro
-comm==0.2.2
-    # via
-    #   ipykernel
-    #   ipywidgets
-contourpy==1.3.2
-    # via matplotlib
-crc32c==2.7.1
-    # via numcodecs
-cycler==0.12.1
-    # via matplotlib
-datasets==3.6.0
-    # via lerobot
-debugpy==1.8.14
-    # via ipykernel
-decorator==5.2.1
-    # via
-    #   gcsfs
-    #   ipython
-deepdiff==8.5.0
-    # via lerobot
-diffusers==0.33.1
-    # via lerobot
-dill==0.3.8
-    # via
-    #   datasets
-    #   multiprocess
-distlib==0.3.9
-    # via virtualenv
-dm-control==1.0.14
-    # via gym-aloha
-dm-env==1.6
-    # via dm-control
-dm-tree==0.1.8 ; python_full_version >= '3.13'
-    # via
-    #   dm-control
-    #   dm-env
-    #   openpi
-    #   openpi-client
-dm-tree==0.1.9 ; python_full_version < '3.13'
-    # via
-    #   dm-control
-    #   dm-env
-    #   openpi
-    #   openpi-client
-docker-pycreds==0.4.0
-    # via wandb
-docstring-parser==0.16
-    # via tyro
-donfig==0.8.1.post1
-    # via zarr
-draccus==0.10.0
-    # via lerobot
-einops==0.8.1
-    # via
-    #   augmax
-    #   lerobot
-    #   openpi
-equinox==0.12.2
-    # via openpi
-etils==1.12.2
-    # via
-    #   optax
-    #   orbax-checkpoint
-evdev==1.9.2 ; 'linux' in sys_platform
-    # via pynput
-executing==2.2.0
-    # via stack-data
-farama-notifications==0.0.4
-    # via gymnasium
-filelock==3.18.0
-    # via
-    #   datasets
-    #   diffusers
-    #   gdown
-    #   huggingface-hub
-    #   openpi
-    #   torch
-    #   transformers
-    #   virtualenv
-flask==3.1.1
-    # via lerobot
-flatbuffers==25.2.10
-    # via openpi
-flax==0.10.2
-    # via openpi
-fonttools==4.58.1
-    # via matplotlib
-frozenlist==1.6.0
-    # via
-    #   aiohttp
-    #   aiosignal
-fsspec==2025.3.0
-    # via
-    #   datasets
-    #   etils
-    #   gcsfs
-    #   huggingface-hub
-    #   openpi
-    #   torch
-gcsfs==2025.3.0
-    # via fsspec
-gdown==5.2.0
-    # via lerobot
-gitdb==4.0.12
-    # via gitpython
-gitpython==3.1.44
-    # via wandb
-glfw==2.9.0
-    # via
-    #   dm-control
-    #   mujoco
-google-api-core==2.24.2
-    # via
-    #   google-cloud-core
-    #   google-cloud-storage
-google-auth==2.40.2
-    # via
-    #   gcsfs
-    #   google-api-core
-    #   google-auth-oauthlib
-    #   google-cloud-core
-    #   google-cloud-storage
-google-auth-oauthlib==1.2.2
-    # via gcsfs
-google-cloud-core==2.4.3
-    # via google-cloud-storage
-google-cloud-storage==3.1.0
-    # via gcsfs
-google-crc32c==1.7.1
-    # via
-    #   google-cloud-storage
-    #   google-resumable-media
-google-resumable-media==2.7.2
-    # via google-cloud-storage
-googleapis-common-protos==1.70.0
-    # via google-api-core
-gym-aloha==0.1.1
-    # via openpi
-gymnasium==0.29.1
-    # via
-    #   gym-aloha
-    #   lerobot
-h5py==3.13.0
-    # via lerobot
-hf-transfer==0.1.9
-    # via huggingface-hub
-hf-xet==1.1.2 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
-    # via huggingface-hub
-huggingface-hub==0.32.3
-    # via
-    #   datasets
-    #   diffusers
-    #   lerobot
-    #   tokenizers
-    #   transformers
-humanize==4.12.3
-    # via orbax-checkpoint
-identify==2.6.12
-    # via pre-commit
-idna==3.10
-    # via
-    #   requests
-    #   yarl
-imageio==2.37.0
-    # via
-    #   gym-aloha
-    #   lerobot
-    #   openpi
-imageio-ffmpeg==0.6.0
-    # via imageio
-iniconfig==2.1.0
-    # via pytest
-inquirerpy==0.3.4
-    # via huggingface-hub
-ipykernel==6.29.5
-ipython==9.2.0
-    # via
-    #   ipykernel
-    #   ipywidgets
-ipython-pygments-lexers==1.1.1
-    # via ipython
-ipywidgets==8.1.7
-itsdangerous==2.2.0
-    # via flask
-jax==0.5.3
-    # via
-    #   augmax
-    #   chex
-    #   equinox
-    #   flax
-    #   openpi
-    #   optax
-    #   orbax-checkpoint
-jax-cuda12-pjrt==0.5.3
-    # via jax-cuda12-plugin
-jax-cuda12-plugin==0.5.3
-    # via jax
+# Core ML/AI packages
+jax[cuda12]==0.5.3
 jaxlib==0.5.3
-    # via
-    #   chex
-    #   jax
-    #   optax
-jaxtyping==0.2.36
-    # via
-    #   equinox
-    #   openpi
-jedi==0.19.2
-    # via ipython
-jinja2==3.1.6
-    # via
-    #   flask
-    #   torch
-jsonlines==4.0.0
-    # via lerobot
-jupyter-client==8.6.3
-    # via ipykernel
-jupyter-core==5.8.1
-    # via
-    #   ipykernel
-    #   jupyter-client
-jupyterlab-widgets==3.0.15
-    # via ipywidgets
-kiwisolver==1.4.8
-    # via matplotlib
-labmaze==1.0.6
-    # via dm-control
-lerobot @ git+https://github.com/huggingface/lerobot@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5
-    # via openpi
-llvmlite==0.44.0
-    # via numba
-lxml==5.4.0
-    # via dm-control
-markdown-it-py==3.0.0
-    # via rich
-markupsafe==3.0.2
-    # via
-    #   flask
-    #   jinja2
-    #   werkzeug
-matplotlib==3.10.3
-matplotlib-inline==0.1.7
-    # via
-    #   ipykernel
-    #   ipython
-mdurl==0.1.2
-    # via markdown-it-py
-mergedeep==1.3.4
-    # via draccus
-ml-collections==1.0.0
-    # via openpi
-ml-dtypes==0.4.1
-    # via
-    #   jax
-    #   jaxlib
-    #   tensorstore
-mpmath==1.3.0
-    # via sympy
-msgpack==1.1.0
-    # via
-    #   flax
-    #   openpi-client
-    #   orbax-checkpoint
-mujoco==2.3.7
-    # via
-    #   dm-control
-    #   gym-aloha
-multidict==6.4.4
-    # via
-    #   aiohttp
-    #   yarl
-multiprocess==0.70.16
-    # via datasets
-mypy-extensions==1.1.0
-    # via typing-inspect
-nest-asyncio==1.6.0
-    # via
-    #   ipykernel
-    #   orbax-checkpoint
-networkx==3.5
-    # via torch
-nodeenv==1.9.1
-    # via pre-commit
-numba==0.61.2
-    # via lerobot
-numcodecs==0.16.1
-    # via zarr
-numpy==1.26.4
-    # via
-    #   augmax
-    #   chex
-    #   contourpy
-    #   datasets
-    #   diffusers
-    #   dm-control
-    #   dm-env
-    #   dm-tree
-    #   flax
-    #   gymnasium
-    #   h5py
-    #   imageio
-    #   jax
-    #   jaxlib
-    #   labmaze
-    #   matplotlib
-    #   ml-dtypes
-    #   mujoco
-    #   numba
-    #   numcodecs
-    #   numpydantic
-    #   opencv-python
-    #   opencv-python-headless
-    #   openpi
-    #   openpi-client
-    #   optax
-    #   orbax-checkpoint
-    #   pandas
-    #   rerun-sdk
-    #   scipy
-    #   tensorstore
-    #   torchvision
-    #   transformers
-    #   treescope
-    #   zarr
-numpydantic==1.6.9
-    # via openpi
-nvidia-cublas-cu12==12.6.4.1 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cudnn-cu12
-    #   nvidia-cusolver-cu12
-    #   torch
-nvidia-cublas-cu12==12.9.0.13 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cudnn-cu12
-    #   nvidia-cusolver-cu12
-nvidia-cuda-cupti-cu12==12.6.80 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-cuda-cupti-cu12==12.9.19 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-cuda-nvcc-cu12==12.9.41
-    # via jax-cuda12-plugin
-nvidia-cuda-nvrtc-cu12==12.6.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-nvidia-cuda-runtime-cu12==12.6.77 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-cuda-runtime-cu12==12.9.37 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-cudnn-cu12==9.5.1.17 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-cudnn-cu12==9.10.1.4 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-cufft-cu12==11.3.0.4 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-cufft-cu12==11.4.0.6 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-cufile-cu12==1.11.1.6 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-nvidia-curand-cu12==10.3.7.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-nvidia-cusolver-cu12==11.7.1.2 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-cusolver-cu12==11.7.4.40 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-cusparse-cu12==12.5.4.2 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cusolver-cu12
-    #   torch
-nvidia-cusparse-cu12==12.5.9.5 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cusolver-cu12
-nvidia-cusparselt-cu12==0.6.3 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-nvidia-ml-py==12.575.51
-    # via pynvml
-nvidia-nccl-cu12==2.26.2 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   torch
-nvidia-nccl-cu12==2.26.5 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via jax-cuda12-plugin
-nvidia-nvjitlink-cu12==12.6.85 ; (platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cufft-cu12
-    #   nvidia-cusolver-cu12
-    #   nvidia-cusparse-cu12
-    #   torch
-nvidia-nvjitlink-cu12==12.9.41 ; (platform_machine == 'aarch64' and sys_platform == 'linux') or sys_platform == 'darwin'
-    # via
-    #   jax-cuda12-plugin
-    #   nvidia-cufft-cu12
-    #   nvidia-cusolver-cu12
-    #   nvidia-cusparse-cu12
-nvidia-nvtx-cu12==12.6.77 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-oauthlib==3.2.2
-    # via requests-oauthlib
-omegaconf==2.3.0
-    # via lerobot
-opencv-python==4.11.0.86
-    # via openpi
-opencv-python-headless==4.11.0.86
-    # via lerobot
-opt-einsum==3.4.0
-    # via jax
-optax==0.2.4
-    # via flax
-orbax-checkpoint==0.11.13
-    # via
-    #   flax
-    #   openpi
-orderly-set==5.4.1
-    # via deepdiff
-packaging==25.0
-    # via
-    #   datasets
-    #   huggingface-hub
-    #   ipykernel
-    #   lerobot
-    #   matplotlib
-    #   pytest
-    #   transformers
-    #   zarr
-pandas==2.2.3
-    # via datasets
-parso==0.8.4
-    # via jedi
-pexpect==4.9.0 ; sys_platform != 'emscripten' and sys_platform != 'win32'
-    # via ipython
-pfzy==0.3.4
-    # via inquirerpy
-pillow==11.2.1
-    # via
-    #   diffusers
-    #   imageio
-    #   matplotlib
-    #   openpi
-    #   openpi-client
-    #   rerun-sdk
-    #   torchvision
-    #   tree
-platformdirs==4.3.8
-    # via
-    #   jupyter-core
-    #   virtualenv
-    #   wandb
-pluggy==1.6.0
-    # via pytest
-polars==1.30.0
-    # via openpi
-pre-commit==4.2.0
-prompt-toolkit==3.0.51
-    # via
-    #   inquirerpy
-    #   ipython
-propcache==0.3.1
-    # via
-    #   aiohttp
-    #   yarl
-proto-plus==1.26.1
-    # via google-api-core
-protobuf==4.25.8
-    # via
-    #   dm-control
-    #   google-api-core
-    #   googleapis-common-protos
-    #   orbax-checkpoint
-    #   proto-plus
-    #   wandb
-psutil==7.0.0
-    # via
-    #   imageio
-    #   ipykernel
-    #   wandb
-ptyprocess==0.7.0 ; sys_platform != 'emscripten' and sys_platform != 'win32'
-    # via pexpect
-pure-eval==0.2.3
-    # via stack-data
-pyarrow==20.0.0
-    # via
-    #   datasets
-    #   rerun-sdk
-pyasn1==0.6.1
-    # via
-    #   pyasn1-modules
-    #   rsa
-pyasn1-modules==0.4.2
-    # via google-auth
-pycparser==2.22
-    # via cffi
-pydantic==2.11.5
-    # via
-    #   numpydantic
-    #   wandb
-pydantic-core==2.33.2
-    # via pydantic
-pygments==2.19.1
-    # via
-    #   ipython
-    #   ipython-pygments-lexers
-    #   rich
-pymunk==7.0.0
-    # via lerobot
-pynput==1.8.1
-    # via lerobot
-pynvml==12.0.0
-pyobjc-core==11.0 ; sys_platform == 'darwin'
-    # via
-    #   pyobjc-framework-applicationservices
-    #   pyobjc-framework-cocoa
-    #   pyobjc-framework-coretext
-    #   pyobjc-framework-quartz
-pyobjc-framework-applicationservices==11.0 ; sys_platform == 'darwin'
-    # via pynput
-pyobjc-framework-cocoa==11.0 ; sys_platform == 'darwin'
-    # via
-    #   pyobjc-framework-applicationservices
-    #   pyobjc-framework-coretext
-    #   pyobjc-framework-quartz
-pyobjc-framework-coretext==11.0 ; sys_platform == 'darwin'
-    # via pyobjc-framework-applicationservices
-pyobjc-framework-quartz==11.0 ; sys_platform == 'darwin'
-    # via
-    #   pynput
-    #   pyobjc-framework-applicationservices
-    #   pyobjc-framework-coretext
-pyopengl==3.1.9
-    # via
-    #   dm-control
-    #   mujoco
-pyparsing==3.2.3
-    # via
-    #   dm-control
-    #   matplotlib
-pysocks==1.7.1
-    # via requests
-pytest==8.3.5
-python-dateutil==2.9.0.post0
-    # via
-    #   jupyter-client
-    #   matplotlib
-    #   pandas
-python-xlib==0.33 ; 'linux' in sys_platform
-    # via pynput
-pytz==2025.2
-    # via pandas
-pywin32==310 ; platform_python_implementation != 'PyPy' and sys_platform == 'win32'
-    # via jupyter-core
-pyyaml==6.0.2
-    # via
-    #   datasets
-    #   donfig
-    #   draccus
-    #   flax
-    #   huggingface-hub
-    #   ml-collections
-    #   omegaconf
-    #   orbax-checkpoint
-    #   pre-commit
-    #   pyyaml-include
-    #   transformers
-    #   wandb
-pyyaml-include==1.4.1
-    # via draccus
-pyzmq==26.4.0
-    # via
-    #   ipykernel
-    #   jupyter-client
-    #   lerobot
-regex==2024.11.6
-    # via
-    #   diffusers
-    #   transformers
-requests==2.32.3
-    # via
-    #   datasets
-    #   diffusers
-    #   dm-control
-    #   gcsfs
-    #   gdown
-    #   google-api-core
-    #   google-cloud-storage
-    #   huggingface-hub
-    #   requests-oauthlib
-    #   transformers
-    #   wandb
-requests-oauthlib==2.0.0
-    # via google-auth-oauthlib
-rerun-sdk==0.23.1
-    # via lerobot
-rich==14.0.0
-    # via
-    #   flax
-    #   openpi
-    #   tyro
-rsa==4.9.1
-    # via google-auth
-ruff==0.11.12
-safetensors==0.5.3
-    # via
-    #   diffusers
-    #   transformers
-scipy==1.15.3
-    # via
-    #   dm-control
-    #   jax
-    #   jaxlib
-sentencepiece==0.2.0
-    # via openpi
-sentry-sdk==2.29.1
-    # via wandb
-setproctitle==1.3.6
-    # via wandb
-shtab==1.7.2
-    # via tyro
-simplejson==3.20.1
-    # via orbax-checkpoint
-smmap==5.0.2
-    # via gitdb
-soupsieve==2.7
-    # via beautifulsoup4
-stack-data==0.6.3
-    # via ipython
-svgwrite==1.4.3
-    # via tree
-sympy==1.14.0
-    # via torch
-tensorstore==0.1.74
-    # via
-    #   flax
-    #   orbax-checkpoint
-termcolor==3.1.0
-    # via lerobot
-tokenizers==0.21.1
-    # via transformers
-toml==0.10.2
-    # via draccus
-toolz==1.0.0
-    # via chex
+jax-cuda12-plugin==0.5.3
+jax-cuda12-pjrt==0.5.3
 torch==2.7.1
-    # via
-    #   lerobot
-    #   openpi
-    #   torchvision
-torchcodec==0.4.0 ; (platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l' and sys_platform == 'linux') or (platform_machine != 'x86_64' and sys_platform == 'darwin') or (sys_platform != 'darwin' and sys_platform != 'linux' and sys_platform != 'win32')
-    # via lerobot
 torchvision==0.22.1
-    # via lerobot
-tornado==6.5.1
-    # via
-    #   ipykernel
-    #   jupyter-client
-tqdm==4.67.1
-    # via
-    #   datasets
-    #   dm-control
-    #   gdown
-    #   huggingface-hub
-    #   tqdm-loggable
-    #   transformers
-tqdm-loggable==0.2
-    # via openpi
-traitlets==5.14.3
-    # via
-    #   comm
-    #   ipykernel
-    #   ipython
-    #   ipywidgets
-    #   jupyter-client
-    #   jupyter-core
-    #   matplotlib-inline
+flax==0.10.2
+optax==0.2.4
 transformers==4.53.2
-    # via openpi
-tree==0.2.4
-    # via openpi-client
-treescope==0.1.9
-    # via openpi
-triton==3.3.1 ; platform_machine == 'x86_64' and sys_platform == 'linux'
-    # via torch
-typeguard==4.4.2
-    # via tyro
-typing-extensions==4.13.2
-    # via
-    #   beautifulsoup4
-    #   chex
-    #   equinox
-    #   etils
-    #   flax
-    #   gymnasium
-    #   huggingface-hub
-    #   ipython
-    #   numcodecs
-    #   openpi
-    #   orbax-checkpoint
-    #   pydantic
-    #   pydantic-core
-    #   rerun-sdk
-    #   torch
-    #   typeguard
-    #   typing-inspect
-    #   typing-inspection
-    #   tyro
-    #   wandb
-    #   zarr
-typing-inspect==0.9.0
-    # via draccus
-typing-inspection==0.4.1
-    # via pydantic
-tyro==0.9.22
-    # via openpi
-tzdata==2025.2
-    # via pandas
-virtualenv==20.31.2
-    # via pre-commit
-wadler-lindig==0.1.6
-    # via equinox
-wandb==0.19.11
-    # via
-    #   lerobot
-    #   openpi
-wcwidth==0.2.13
-    # via prompt-toolkit
-websockets==15.0.1
-    # via openpi-client
-werkzeug==3.1.3
-    # via flask
-widgetsnbextension==4.0.14
-    # via ipywidgets
-wrapt==1.14.1 ; python_full_version < '3.13'
-    # via dm-tree
-xxhash==3.5.0
-    # via datasets
-yarl==1.20.0
-    # via aiohttp
-zarr==3.0.8
-    # via lerobot
+diffusers==0.33.1
+
+# Computer vision and image processing
+opencv-python==4.11.0.86
+pillow>=11.0.0
+imageio>=2.36.1
+
+# Scientific computing (core)
+numpy>=1.22.4,<2.0.0
+scipy>=1.15.0
+
+# Data handling
+dm-tree>=0.1.8
+einops>=0.8.0
+pandas>=2.0.0
+polars>=1.30.0
+
+# ML utilities
+ml-collections==1.0.0
+orbax-checkpoint==0.11.13
+sentencepiece>=0.2.0
+datasets>=3.0.0
+huggingface-hub>=0.20.0
+
+# Project specific
+augmax>=0.3.4
+equinox>=0.11.8
+jaxtyping==0.2.36
+numpydantic>=1.6.6
+beartype==0.19.0
+treescope>=0.1.7
+
+# File handling and storage
+fsspec[gcs]>=2024.6.0
+flatbuffers>=24.3.25
+h5py>=3.10.0
+
+# Utilities
+tqdm>=4.60.0
+rich>=14.0.0
+tyro>=0.9.5
+wandb>=0.19.1
+filelock>=3.16.1
+
+# Web and networking
+requests>=2.25.0
+aiohttp>=3.8.0
+websockets>=11.0
+
+# Development (minimal - remove if not needed)
+typing-extensions>=4.12.2
+
+# Robotics/Control specific
+gymnasium>=0.29.0
+mujoco>=2.3.0