#JAX

オープンソース

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

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

オープンソース

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

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