PyPI page
Home page
Author:
None
Summary:
Modular imitation learning framework for training, compressing, and deploying robot manipulation policies
Latest version:
0.4.1
Required dependencies:
accelerate
|
albumentations
|
av
|
diffusers
|
executorch
|
geomloss
|
google-crc32c
|
h5py
|
huggingface-hub
|
hydra-core
|
joblib
|
lightning
|
matplotlib
|
numba
|
numpy
|
omegaconf
|
opencv-python-headless
|
pandas
|
peft
|
psutil
|
pyarrow
|
pykeops
|
python-dotenv
|
pytorch-lightning
|
pyyaml
|
safetensors
|
scikit-learn
|
scipy
|
seaborn
|
termcolor
|
threadpoolctl
|
timm
|
torch
|
torchao
|
torchvision
|
tqdm
|
transformers
|
tso-robotics-sockets
|
versatil-constants
|
wandb
|
zarr
Optional dependencies:
mkdocs-gen-files
|
mkdocs-literate-nav
|
mkdocs-material
|
mkdocstrings
|
torch
|
torchao
|
torchvision
Downloads last day:
13
Downloads last week:
183
Downloads last month:
196