JAX - 资源平台：AI工具、网站、APP和开源项目合集
llama-2-jax
Llama 2JAX
基于JAX的大语言模型高效实现
QDax
QDax质量多样性
高效加速质量多样性算法的开源框架
pgx
Pgx强化学习
JAX原生并行游戏模拟器库用于强化学习研究
skrl
SKRL强化学习
开源模块化强化学习库
seqax
JAXseqax
JAX序列建模框架简化小型到中型LLM预训练研究
dynamax
状态空间模型JAX
JAX驱动的概率状态空间模型库
optax
OptaxJAX
JAX生态系统中的高效梯度处理与优化框架
vision_transformer
Vision TransformerMLP-Mixer
视觉Transformer和MLP-Mixer模型库 高性能图像识别
flax
FlaxJAX
灵活强大的JAX神经网络库和生态系统
rlax
RLax强化学习
JAX驱动的强化学习算法构建库
dinosaur
Dinosaur大气建模
全球大气建模的新工具 可微分动力学谱系核心
lineax
Lineax线性求解
基于JAX的线性求解和最小二乘优化库
gemma
Gemma大语言模型
Google DeepMind开源的Gemma大语言模型
jax-triton
JAXTriton
JAX与Triton集成实现GPU计算加速
flashbax
Flashbax经验回放缓冲区
JAX强化学习高效体验回放缓冲库
Mava
Mava多智能体强化学习
基于JAX的高效多智能体强化学习框架
scenic
Scenic计算机视觉
多模态视觉智能研究框架
jax
JAX自动微分
高性能科学计算和机器学习的Python加速库
s2fft
S2FFT球谐变换
基于JAX和PyTorch的球谐变换Python库
levanter
Levanter大语言模型
专注可读性与可扩展性的大语言模型训练框架
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号