xLSTM-jax: JAX 实现的 xLSTM 模型库,提供高效的训练和推理代码

xLSTM-jax: JAX 实现的 xLSTM 模型库,提供高效的训练和推理代码

【xLSTM-jax: JAX 实现的 xLSTM 模型库,提供高效的训练和推理代码】'NX-AI/xlstm-jax: Official JAX implementation of xLSTM incl

2024-12-16浏览详情

Zephyr:一个基于JAX的声明式神经网络库,让设计、创建和操作神经网络变得更简单快捷,特别适合想要快速实现机器学习想法的开发者

Zephyr:一个基于JAX的声明式神经网络库,让设计、创建和操作神经网络变得更简单快捷,特别适合想要快速实现机器学习想法的开发者

【Zephyr:一个基于JAX的声明式神经网络库,让设计、创建和操作神经网络变得更简单快捷,特别适合想要快速实现机器学习想法的开发者】'

2024-12-11浏览详情

Jax2D:一个用JAX编写的高可扩展2D物理引擎,能够快速模拟多样的小规模物理场景

Jax2D:一个用JAX编写的高可扩展2D物理引擎,能够快速模拟多样的小规模物理场景

【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-flash-attn2:一个为JAX设计的灵活高效的Flash Attention 2.0实现,支持GPU/TPU/CPU多后端和Triton/Pallas/JAX多平台

【jax-flash-attn2:一个为JAX设计的灵活高效的Flash Attention 2.0实现,支持GPU/TPU/CPU多后端和Triton/Pallas/JAX多平台】'jax-fl

2024-10-27浏览详情

JAX机器学习栈:基于JAX的AI/ML技术栈

JAX机器学习栈:基于JAX的AI/ML技术栈

【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: JAX写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化

【JAX-Fluids: JAX
写的可微流体动力学包,运行在 CPU/GPU/TPU 上,支持自动微分对数值模型进行端到端优化】’JAX-Fluids: A Differe

2022-11-07浏览详情

正在拼命加载中

我是有底线的

没有更多的页面可以加载啦!