TensorFlow Multi-GPU Batch Text Generation
Configures a distributed text generation pipeline using TensorFlow MirroredStrategy and Hugging Face Transformers, handling specific tokenizer padding requirements and batch processing logic.
npx skills add ECNU-ICALK/AutoSkill --skill tensorflow-multi-gpu-batch-text-generation --agent claude-code
Same command for any agent — swap --agent for codex, cursor, copilot.
Weekly change comes from our own snapshots, not the repository page — it measures attention, not adoption.
# TensorFlow Multi-GPU Batch Text Generation Configures a distributed text generation pipeline using TensorFlow MirroredStrategy and Hugging Face Transformers, handling specific tokenizer padding requirements and batch processing logic. ## Prompt # Role & Objective You are a Machine Learning Engineer specializing in TensorFlow and Hugging Face Transformers. Your task is to implement a distributed text generation pipeline using `tf.distribute.MirroredStrategy` for multi-GPU inference. # Operational Rules & Constraints 1. **Strategy Initialization**: Initialize `tf.distribute.MirroredStrategy` with the specific GPU devices requested (e.g., `["/gpu:0", "/gpu:1", ...]`). 2. **Model Loading**: Load `TFAutoModelForCausalLM` and `AutoTokenizer` inside the `strategy.scope()`. 3. **Tokenizer Configuration**: Explicitly set the padding token to prevent errors for models like GPT-2: `tokenizer.pad_token = tokenizer.eos_token`. 4. **Batch Processing**: Implement a function (e.g., `generate_response`) that accepts `context_messages` and `user_prompts`. Combine these into a list of strings for batch processing. 5. **Tokenization**: Use `tokenizer(..., return_tensors='tf', padding=True, truncatio
- Prompt
- Triggers
What does the TensorFlow Multi-GPU Batch Text Generation skill do?
Configures a distributed text generation pipeline using TensorFlow MirroredStrategy and Hugging Face Transformers, handling specific tokenizer padding requirements and batch processing logic.
How do I install it?
Run `npx skills add ECNU-ICALK/AutoSkill --skill tensorflow-multi-gpu-batch-text-generation --agent claude-code` — it drops the skill into your project so the agent can pick it up. Swap the --agent value for codex, cursor or copilot if you use one of those.
Where does this skill come from?
From ECNU-ICALK/AutoSkill, a repository with 539 stars. We read it straight from the repository tree rather than a submitted listing, so what you see here is what is actually published.
Is a popular skill a good skill?
Not necessarily. Stars measure attention, not adoption — a repository can trend for a week and be abandoned. That is why we show the weekly change from our own snapshots next to the total, instead of a single flattering number.