trtutils.research package¶

Subpackages¶

trtutils.research.axonn package
- Module contents

Module contents¶

Implementations for research papers.

Submodules¶

axonn: Implementation of the AxoNN paper for energy-aware multi-accelerator neural network inference optimization.

trtutils.research.discover_submodules() → list[str][source]¶

Return a list of importable research submodule names.

Returns:: A list of importable research submodule names.
Return type:: list[str]

trtutils.research.register_cli(subparsers: argparse._SubParsersAction, parents: list[argparse.ArgumentParser]) → None[source]¶

Register CLI subcommands for all available research submodules.

Parameters:

subparsers (argparse._SubParsersAction) – The subparsers action to register the CLI commands under.
parents (list[argparse.ArgumentParser]) – The parent parsers to use for the CLI commands.