3,616 repos GH 3,501 / HF 115 · 11 categories · GitHub上のAI関連OSSを日本語で早見 · 毎日自動更新

#JAX (2 repos)

「JAX」タグが付いたリポジトリ

← 全リポジトリ

ASR / 音声認識
GitHub
whisper-jax
Whisper JAXは、OpenAIのWhisperモデルをJAXで実装したもので、特にTPUでの実行時にOpenAIのPyTorch版と比較して最大…
#CLI #GPU対応 #Hugging Face
★ 4.7k ⑂ 414 Apache-2.0 2024-04-03
LLM
GitHub
jax-7b-tpu-sharding
本リポジトリは、JAX/Flax/Optaxを用いてTPU v4-8上で7BパラメータのTransformerモデルをトレーニングする際に遭遇す…
#CLI #JAX #Python
★ 1 2026-05-13