0 Kommentare (0 Kommentare)0 Reaktionen (0 Reaktionen)1 zugewiesene Person (1 zugewiesene Person)Python101 Forks (101 Forks)auto 404
help wantedperformance
Repository-Metriken
- Stars
- 276 Sterne (276 Sterne)
- PR-Merge-Metriken
- Durchschn. Merge 7T 7h (Durchschn. Merge 7T 7h)34 gemergte PRs in 30 T (34 gemergte PRs in 30 T)
Beschreibung
- model: Qwen/Qwen3-235B-A22B
- chip: v6e
- input: [4, 64, 256, 512, 1024, 2048, 4096]
- output: [1, 4, 1024, 4096, 8192, 16384, 24576]
- batch_size: [8,16,32,64,128,256,512]
- mode: McJAX
- hardware: tpu-v6e-32
Contributor Guide
- Research-Richtung
- Richten Sie die Umgebung ein, um das Modell Qwen/Qwen3-235B A22B auf TPU v6e mit JAX auszuführen, führen Sie die angegebenen Eingabeformen und Batch Größen aus, messen Sie Latenz und Durchsatz und vergleichen Sie sie mit den erwarteten Leistungsbaselines.
- Tech Stack
- python
- Domain
- machine learning
- Issue Type
- Test
- Voraussetzungen
- PythonJAXMachine Learning basics