Skip to content

Commit 772f8c5

Browse files
gvanicaclaude
andcommitted
Overhaul migration prompts with JAX best practices and MaxText support
Rewrites and expands the migration prompt library with: - Comprehensive JAX/Flax best practices (KV cache, causal conv1d, static shapes, attention masking, rotary embeddings, etc.) - Single-file and multi-file conversion prompts - MaxText-specific conversion prompts (YAML overlay, custom layers) - Hallucination prevention rules - Flax scoping and naming conventions Also updates models.py with additional model configuration support. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent a998c34 commit 772f8c5

2 files changed

Lines changed: 434 additions & 39 deletions

File tree

0 commit comments

Comments
 (0)