#JAX

オープンソース

NVIDIA SOLARでPyTorch性能の理論下限を自動推定

GPU最適化の前に「このモデルは理論上どこまで速くなれるか」…

オープンソース

PTXカーネルをPythonで手書き—pyptxの使い方と実測値

GPUカーネルを完全に制御したい場合、最終的にはPTX（Pa…

オープンソース

FlashInferをJAXで動かすjax-tvm-ffiの使い方

JAXはLLMの学習に強いフレームワークです。ただし推論フェ…