多码网
返回 AI
AIlibraryJupyter Notebook

get-started-with-JAX

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

781 stars119 forks1 issuesMIT更新于 2026/4/23
deep-learningflaxhaikujaxjupyterlaxlearn-jaxmachine-learningnumpyoptaxpythontutorialxla
暂无详细内容

同类推荐