Home

Awesome

<br /> <p align="center"> <!-- Note that this README will be used for both GitHub and PyPI. We therefore: - Keep all image URLs absolute. - In the GitHub action we use for publishing, strip some HTML tags that aren't supported by PyPI. --> <!-- pypi-strip --> <picture> <source media="(prefers-color-scheme: dark)" srcset="https://brentyi.github.io/tyro/_static/logo-dark.svg" /> <!-- /pypi-strip --> <img alt="tyro logo" src="https://brentyi.github.io/tyro/_static/logo-light.svg" width="200px" /> <!-- pypi-strip --> </picture> <!-- /pypi-strip --> </p> <p align="center"> <em><a href="https://brentyi.github.io/tyro">Documentation</a></em> &nbsp;&nbsp;&bull;&nbsp;&nbsp; <em><code>pip install tyro</code></em> </p> <p align="center"> <img alt="build" src="https://github.com/brentyi/tyro/actions/workflows/build.yml/badge.svg" /> <img alt="mypy" src="https://github.com/brentyi/tyro/actions/workflows/mypy.yml/badge.svg" /> <img alt="pyright" src="https://github.com/brentyi/tyro/actions/workflows/pyright.yml/badge.svg" /> <img alt="ruff" src="https://github.com/brentyi/tyro/actions/workflows/ruff.yml/badge.svg" /> <a href="https://codecov.io/gh/brentyi/tyro"> <img alt="codecov" src="https://codecov.io/gh/brentyi/tyro/branch/main/graph/badge.svg" /> </a> <a href="https://pypi.org/project/tyro/"> <img alt="codecov" src="https://img.shields.io/pypi/pyversions/tyro" /> </a> </p> <br />

<strong><code>tyro.cli()</code></strong> is a tool for generating CLI interfaces.

We can define configurable scripts using functions:

"""A command-line interface defined using a function signature.

Usage: python script_name.py --foo INT [--bar STR]
"""

import tyro

def main(
    foo: int,
    bar: str = "default",
) -> None:
    ...  # Main body of a script.

if __name__ == "__main__":
    # Generate a CLI and call `main` with its two arguments: `foo` and `bar`.
    tyro.cli(main)

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

"""A command-line interface defined using a class signature.

Usage: python script_name.py --foo INT [--bar STR]
"""

from dataclasses import dataclass
import tyro

@dataclass
class Config:
    foo: int
    bar: str = "default"

if __name__ == "__main__":
    # Generate a CLI and instantiate `Config` with its two arguments: `foo` and `bar`.
    config = tyro.cli(Config)

    # Rest of script.
    assert isinstance(config, Config)  # Should pass.

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

In the wild

tyro is designed to be lightweight enough for throwaway scripts, while facilitating type safety and modularity for larger projects. Examples:

<table> <tr> <td> <a href="https://github.com/nerfstudio-project/nerfstudio/"> nerfstudio-project/nerfstudio <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/nerfstudio-project/nerfstudio?style=social" /> </a> </td> <td> Open-source tools for neural radiance fields. </td> </tr> <tr> <td> <a href="https://github.com/Sea-Snell/JAXSeq/"> Sea-Snell/JAXSeq <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/Sea-Snell/JAXSeq?style=social" /> </a> </td> <td>Train very large language models in Jax.</td> </tr> <tr> <td> <a href="https://github.com/kevinzakka/obj2mjcf"> kevinzakka/obj2mjcf <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/kevinzakka/obj2mjcf?style=social" /> </a> </td> <td>Interface for processing OBJ files for Mujoco.</td> </tr> <tr> <td> <a href="https://github.com/blurgyy/jaxngp"> blurgyy/jaxngp <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/blurgyy/jaxngp?style=social" /> </a> </td> <td> CUDA-accelerated implementation of <a href="https://nvlabs.github.io/instant-ngp/">instant-ngp</a>, in JAX. </td> </tr> <tr> <td> <a href="https://github.com/NVIDIAGameWorks/kaolin-wisp"> NVIDIAGameWorks/kaolin-wisp <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/NVIDIAGameWorks/kaolin-wisp?style=social" /> </a> </td> <td>PyTorch library for neural fields.</td> </tr> <tr> <td> <a href="https://github.com/autonomousvision/sdfstudio"> autonomousvision/sdfstudio <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/autonomousvision/sdfstudio?style=social" /> </a> </td> <td>Unified framework for surface reconstruction.</td> </tr> <tr> <td> <a href="https://github.com/openrlbenchmark/openrlbenchmark"> openrlbenchmark/openrlbenchmark <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/openrlbenchmark/openrlbenchmark?style=social" /> </a> </td> <td>Collection of tracked experiments for reinforcement learning.</td> </tr> <tr> <td> <a href="https://github.com/vwxyzjn/cleanrl"> vwxyzjn/cleanrl <br /><img alt="GitHub star count" src="https://img.shields.io/github/stars/vwxyzjn/cleanrl?style=social" /> </a> </td> <td>Single-file implementation of deep RL algorithms.</td> </tr> </table>

Alternatives

tyro is an opinionated library. 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 particularly like:

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