Skip to content

brentyi/tyro

Repository files navigation

tyro

codecov PyPI Downloads codecov

Documentation   •   pip install tyro

tyro.cli() is a tool for generating CLI interfaces from type-annotated Python.

We can define configurable scripts using functions:

tyro_function.mp4

Or instantiate configs defined using tools like dataclasses, pydantic, and attrs:

tyro_dataclass.mp4

Other features include helptext generation, nested structures, subcommands, and shell completion. For examples and the API reference, see our documentation.

Why tyro?

  1. Define things once. Standard Python type annotations, docstrings, and default values are parsed to automatically generate command-line interfaces with nice helptext.

  2. Static types. Unlike tools dependent on dictionaries, YAML, or dynamic namespaces, arguments populated by tyro are better undestood by IDEs and language servers, as well as static checking tools like pyright and mypy.

  3. Modularity. tyro supports hierarchical configurations, which make it easy to decentralize definitions, defaults, and documentation.

In the wild

tyro is designed to be lightweight for throwaway scripts, while improving maintainability for larger projects. Examples:

nerfstudio-project/nerfstudio
GitHub stars
Open-source tools for neural radiance fields.
Sea-Snell/JAXSeq
GitHub stars
Train very large language models in Jax.
kevinzakka/obj2mjcf
GitHub stars
Interface for processing OBJ files for Mujoco.
blurgyy/jaxngp
GitHub stars
CUDA-accelerated implementation of instant-ngp, in JAX.
NVIDIAGameWorks/kaolin-wisp
GitHub stars
PyTorch library for neural fields.
autonomousvision/sdfstudio
GitHub stars
Unified framework for surface reconstruction.
openrlbenchmark/openrlbenchmark
GitHub stars
Collection of tracked experiments for reinforcement learning.
vwxyzjn/cleanrl
GitHub stars
Single-file implementation of deep RL algorithms.
pytorch-labs/LeanRL
GitHub stars
Fork of CleanRL, optimized using PyTorch 2 features.
pytorch/torchtitan
GitHub stars
PyTorch-native platform for training generative AI models.
KwaiVGI/LivePortrait
GitHub stars
Stitching and retargeting for portraits.
Physical-Intelligence/openpi
GitHub stars
Open-source models for robotics.
mujocolab/mjlab
GitHub stars
Lightweight, modular abstractions for RL and sim-to-real robotics.
amazon-far/holosoma
GitHub stars
Humanoid robotics framework for RL training and deployment.
MalcolmMielle/bark_monitor
GitHub stars
Show your neighbor that your dog doesn't bark!

Alternatives

tyro is opinionated. If any design decisions don't make sense, feel free to file an issue!

You might also consider one of many alternative libraries. Some that we like:

  • cappa offers a similar core feature set but with very different ergonomics. It looks polished and well-maintained!
  • cyclopts and defopt has very comprehensive type annotation support and a heavier emphasis on subcommand generation.
  • simple-parsing and jsonargparse provide deeper integration with configuration file formats like YAML and JSON.
  • clipstick, which focuses on simplicity + generating CLIs from Pydantic models.
  • datargs provides a minimal API for dataclasses.
  • fire and clize support arguments without type annotations.

There are also some options that directly extend tyro:

  • mininterface simultaneously generates GUI, TUI, web, CLI, and file-based program configuration.
  • manuscript generates CLI interfaces from a simple block of configuration variables.

We also have some notes on tyro's design goals and other alternatives in the docs here.