★ 3
Pearl-Chat：用 JAX 从零构建 850 万参数卢干达语模型并开源
论文发布 Pearl-Chat，一个约 847 万参数的因果解码器 Transformer，专为东非逾千万人使用的低资源班图语卢干达语设计，基于 JAX 与 Flax NNX 从第一性原理实现。自建字节级 BPE 分词器（词表 8192）实现每 token 4.88 字符、每词 1.60 toke…
arXiv NLP14 天前
#JAX
共 5 篇相关资讯，按重要度与时间排序，每 3 小时更新。
★ 2
Jaxnasium 开源：PyTrees 统一 JAX 强化学习跨环境套件
研究者开源 Jaxnasium，用 PyTrees 表示观测与动作空间，使同一份算法代码可运行于离散、连续、图像、复合及多智能体环境。库接入 11 个环境套件、300 多个环境，并提供多随机种子评估与超参扫描工具。
arXiv cs.LG5 天前
★ 2
论文提出无坐标自动微分算子
论文指出主流自动微分框架隐式固定欧氏空间基，难以表达旋转、体积形式、曲率等几何对象。作者提出单一与基无关的导数算子，可表示多种几何微分运算，且完全用现有 AD 原语表达，可与 JAX 批处理、编译与硬件加速组合，并在流形算子、时空反问题管线与转子子群优化中验证复用性。
arXiv cs.LG21 天前
★ 2
JaxAHT：基于JAX的即兴协作库
即兴协作（AHT）研究受制于高昂计算成本、缺乏标准基准和评估队友套件。作者推出首个开源JAX库JaxAHT，统一队友生成、自我智能体训练与评估流程，相比PyTorch实现约95倍墙钟加速。该库还提供Level-Based Foraging、Overcooked和Hanabi的评估队友套件，并用于大…
arXiv cs.AI48 天前
★ 2
基于JAX3D的分层NeRF体渲染与三维重建教程
该教程使用JAX、Flax、Optax及jax3d提供的体渲染原语，构建端到端的分层神经辐射场（NeRF）。作者先从解析场景生成含体积几何与视角相关辐射的多视角合成数据集，再实现分层NeRF以提升渲染与重建质量。内容面向希望掌握新视角合成与三维重建工程实现的开发者。
MarkTechPost49 天前