TensorFlow, PyTorch が有名ですが，最近は JAX をよく見かけるようになりました．
論文で使用されているフレームワークのシェアを確認すると，PyTorchが圧勝です．TensorFlowが使われる頻度はかなり減っているようです．JAXやMindSporeは徐々に増えているようです．
Hugging Face のライブラリ別のモデル数を比較してみると，やはり PyTorch が圧勝しています．
最近は生成系 AI が進化していて，meta Llama 2 や microsoft phi-1 など次々と素晴らしいモデルが発表されています．これらは PyTorch を利用しているようです．
したがって，特別な理由がなければ PyTorch を選択するのがよさそうです．
