LLM GitHub jax-7b-tpu-sharding本リポジトリは、JAX/Flax/Optaxを用いてTPU v4-8上で7BパラメータのTransformerモデルをトレーニングする際に遭遇す… #CLI #JAX #Python ★ 1 2026-05-13