PyPI page
Home page
Author:
None
License:
MIT
Summary:
A python library for training any robot policy through Imitation Learning, with dependencies handled by uv
Latest version:
0.3.0
Required dependencies:
accelerate
|
albumentations
|
av
|
diffusers
|
geomloss
|
h5py
|
huggingface-hub
|
hydra-core
|
joblib
|
lightning
|
matplotlib
|
numba
|
numpy
|
omegaconf
|
opencv-python-headless
|
pandas
|
pre-commit
|
psutil
|
pyarrow
|
pykeops
|
pytest
|
python-dotenv
|
pytorch-lightning
|
pyyaml
|
ruff
|
safetensors
|
scikit-learn
|
scipy
|
seaborn
|
termcolor
|
threadpoolctl
|
timm
|
torch
|
torchao
|
torchaudio
|
torchvision
|
tqdm
|
transformers
|
triton
|
tso-robotics-sockets
|
versatil-constants
|
wandb
|
zarr
Optional dependencies:
mkdocs-gen-files
|
mkdocs-literate-nav
|
mkdocs-material
|
mkdocstrings
Downloads last day:
1
Downloads last week:
13
Downloads last month:
108