What is the primary purpose of this skill?

It helps developers export, optimize, and deploy TensorFlow and JAX models to various production environments efficiently.

What quantization techniques are covered?

It covers post-training float16 quantization, dynamic range quantization, and full integer quantization for maximum model compression.

Does it help with TPU optimization?

Yes, it includes patterns for MirroredStrategy and TPU variable reformatting to optimize distributed training and serving.

Does it support mobile deployment?

Yes, it includes comprehensive patterns for TFLite conversion, Android deployment, and mobile-specific graph optimizations.

Can I use this for JAX models?

Yes, the skill provides specific workflows using Orbax and JaxModule to convert JAX code to SavedModel and TFLite formats.

TensorFlow Model Deployment

Name: TensorFlow Model Deployment
Author: TheBushidoCollective

byTheBushidoCollective

•

Ciencia de Datos y ML

Deploys, optimizes, and converts TensorFlow models for production environments including mobile, edge, and cloud.

This skill provides specialized guidance for transitioning TensorFlow and JAX models from research to production. It streamlines the workflow for exporting SavedModels, performing TFLite conversion, and applying advanced quantization techniques like float16 and full integer quantization. By offering optimized code patterns for graph transformations and serving infrastructure, it enables developers to deploy high-performance machine learning models efficiently across Android devices, cloud servers, and TPU-powered environments.

Características Principales

01SavedModel export with custom serving signatures

02JAX to TensorFlow and TFLite model conversion workflows

03Post-training quantization (FP16, Dynamic Range, and Full Integer)

0474 GitHub stars

05TFLite conversion and optimization for mobile and edge devices

06Graph transformations to optimize models for inference speed

Casos de Uso

01Migrating research-based JAX models into production-ready SavedModel formats

02Converting Keras models into compressed TFLite formats for mobile applications

03Deploying EfficientDet object detection models with optimized serving signatures

What are Skills?·How to Install

Install with 🐟 Skill.Fish

npx skillfish add thebushidocollective/han tensorflow-model-deployment

For use in Claude.ai and ChatGPT

Download Skill