JAX - 资源平台：AI工具、网站、APP和开源项目合集
XLB
XLBLattice Boltzmann
基于JAX的可微分格子玻尔兹曼方法库
jaxdf
jaxdfJAX
JAX框架打造可微分物理模拟器
synjax
SynJaxJAX
基于JAX的结构化概率分布神经网络库
evojax
EvoJAX神经进化
基于JAX的高性能神经进化工具包
evosax
evosax进化策略
基于JAX的高性能进化策略框架
blackjax
BlackJAXJAX
JAX贝叶斯采样库 支持CPU和GPU运算
brax
Brax物理引擎
基于JAX的高性能物理引擎 适用于机器人和强化学习仿真
jaxlie
jaxlieLie群
JAX Lie群库为计算机视觉和机器人应用提供刚体变换
tf2jax
TF2JAXTensorFlow
实验性TensorFlow到JAX函数转换库
gymnax
gymnax强化学习
JAX驱动的高效强化学习环境集合
mctx
MctxJAX
高效JAX实现的蒙特卡洛树搜索库
learned_optimization
learned_optimization元学习
基于JAX的元学习优化器研究框架
jumanji
Jumanji强化学习
JAX驱动的多样化强化学习环境套件 加速研究与应用
optimistix
OptimistixJAX
JAX生态系统中的高效非线性求解器
jaxtyping
jaxtyping类型注解
JAX数组与PyTrees的类型注解和运行时检查工具
diffrax
DiffraxJAX
JAX 自动微分与 GPU 支持的数值微分方程解析工具
awesome-jax
JAX机器学习
自动微分与XLA在高性能机器学习中的应用
axlearn
AXLearnJAX
支持构建大规模深度学习模型的高效工具库
equinox
EquinoxJAX
强大且易用的JAX兼容神经网络库
dm-haiku
JAXHaiku
JAX神经网络构建的简洁解决方案
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号