| # Core project dependencies - minimal set to avoid system package conflicts |
| # Based on openpi pyproject.toml dependencies |
|
|
| # Core ML/AI packages |
| jax[cuda12]==0.5.3 |
| jaxlib==0.5.3 |
| jax-cuda12-plugin==0.5.3 |
| jax-cuda12-pjrt==0.5.3 |
| flax==0.10.2 |
| optax==0.2.4 |
| diffusers==0.33.1 |
|
|
| # Computer vision and image processing |
| opencv-python==4.11.0.86 |
| # pillow>=11.0.0 # Removed - system package conflict |
| imageio>=2.36.1 |
|
|
| # Scientific computing (core) |
| numpy>=1.22.4,<2.0.0 |
| # scipy>=1.15.0 # Removed - system package conflict |
|
|
| # Data handling |
| dm-tree>=0.1.8 |
| msgpack>=1.0.5 |
| 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 |
|
|
| # Local packages |
| -e ./openpi/packages/openpi-client |
|
|
| # File handling and storage |
| # fsspec[gcs]>=2024.6.0 # Removed - system package conflict |
| flatbuffers>=24.3.25 |
| h5py>=3.10.0 |
|
|
| # Utilities |
| tqdm>=4.60.0 |
| tqdm-loggable>=0.1.0 |
| tree>=0.2.4 |
| 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 # Removed - system package conflict |
| pytest>=6.0.0 |
|
|
| # Robotics/Control specific |
| gymnasium>=0.29.0 |
| mujoco>=2.3.0 |
|
|
| # Remove problematic system packages: |
| # - ipython (use system version or install manually) |
| # - matplotlib (use system version) |
| # - jupyter packages (install separately if needed) |
| # - blinker (system package) |
| # - setuptools/pip related packages |
