PyTorch Gallery

Examples about exporting PyTorch models to ONNX, model patching, and input observation.

Applying patches to a model and displaying the diff

Applying patches to a model and displaying the diff

Export a LLM with InputObserver (with Tiny-LLM)

Export a LLM with InputObserver (with Tiny-LLM)

InputObserver with Transformers Cache

InputObserver with Transformers Cache

InputObserver: recording inputs for ONNX export

InputObserver: recording inputs for ONNX export

Registering a custom class as a pytree node

Registering a custom class as a pytree node