【xLSTM-jax: JAX 实现的 xLSTM 模型库,提供高效的训练和推理代码】'NX-AI/xlstm-jax: Official JAX implementation of xLSTM incl
2024-12-16浏览详情
【Zephyr:一个基于JAX的声明式神经网络库,让设计、创建和操作神经网络变得更简单快捷,特别适合想要快速实现机器学习想法的开发者】'
2024-12-11浏览详情
【Jax2D:一个用JAX编写的高可扩展2D物理引擎,能够快速模拟多样的小规模物理场景】'Jax2D is a highly scalable 2D physics engine
2024-11-17浏览详情
【jax-flash-attn2:一个为JAX设计的灵活高效的Flash Attention 2.0实现,支持GPU/TPU/CPU多后端和Triton/Pallas/JAX多平台】'jax-fl
2024-10-27浏览详情
【JAX机器学习栈:基于JAX的AI/ML技术栈】'jax-ml-stack - Your JAX Powered AI/ML Stack' GitHub: github.com/jax-ml/jax_ml_stac
2024-08-29浏览详情
【JAX-Fluids: JAX 写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化】’JAX-Fluids: A Differe
2022-11-07浏览详情
正在拼命加载中
我是有底线的
没有更多的页面可以加载啦!