executing-plans
executing-plans
Use when executing an implementation plan in the current session as the implementer yourself — your human partner chose inline execution, or no subagent tool is available
nvidia-tilegym-monkey-patch-kernels-to-transformersIntegrate TileGym kernels into Hugging Face `transformers` models by replacing the library's submodule(s) and certain class(es)' implementations, and patching certain class(es)' init/forward/load weight methods prior to instantiating models. Used when the user requires integrating TileGym kernels into `transformers` models.
Independent GitHub snapshot · Automated classification · Original documentation governs
name: nvidia-tilegym-monkey-patch-kernels-to-transformers
description: Integrate TileGym kernels into Hugging Face transformers models by
replacing the library's submodule(s) and certain class(es)' implementations, and
patching certain class(es)' init/forward/load weight methods prior to instantiating
models. Used when the user requires integrating TileGym kernels into transformers
models.
license: CC-BY-4.0 AND Apache-2.0
metadata:
author: TileGym Team TileGym@nvidia.com
tags:
The main purpose of TileGym project is to provide performant kernels for LLM training and inference. We will integrate proper kernels available in TileGym project to LLM models provided by Hugging Face transformers library to validate end-to-end functional correctness and performance improvements. Instead of modifying transformers source code, we will take a non-intrusive monkey-patch approach: We will replace certain modules/classes/methods in transformers library that implement the Transformer model we would like to integrate, such that at model instantiation, that model's core components will be replaced by TileGym implementations. At runtime the model will actually invoke TileGym kernels under the hood. In addition, we will follow an auto-research-style agent harness loop to create and integrate new cuTile kernels to the target model to improve kernel coverage and end-to-end throughput.