google-ai-edge/litert-torch

A tool for converting ONNX files to LiteRT/TFLite/TensorFlow, PyTorch native code (nn.Module), TorchScript (.pt), state_dict (.pt), Exported Program (.pt2), and Dynamo ONNX. It also supports direct conversion from LiteRT to PyTorch.

PyTorch Neural Network eXchange

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

A pytorch quantization backend for optimum

PyTorch native quantization for training and inference

LiteRT and LiteRT-LM sample apps, model recipes, agent skills and utilities.

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX

PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

Serve, optimize and scale PyTorch models in production