小红花·文摘
  • 首页
  • 广场
  • 排行榜🏆
  • 直播
  • FAQ
Dify.AI
谷歌Metrax为JAX带来了预定义的模型评估指标

Google最近开源了Metrax,这是一个JAX库,提供分类、回归、NLP、视觉和音频模型的标准化性能指标,支持分布式和大规模训练,确保指标实现符合最佳实践。

谷歌Metrax为JAX带来了预定义的模型评估指标

InfoQ
InfoQ · 2025-12-20T15:00:00Z
vLLM TPU:支持PyTorch和JAX的新统一后端

vLLM TPU通过tpu-inference插件整合JAX和PyTorch,显著提升性能和模型支持。新设计优化了TPU性能,支持多种模型,简化开发流程。RPA v3内核增强灵活性和效率,SPMD模型提升性能,整体性能较2025年原型提升近5倍,推动开源TPU推理的发展。

vLLM TPU:支持PyTorch和JAX的新统一后端

vLLM Blog
vLLM Blog · 2025-10-16T00:00:00Z
从开发者的角度比较深度学习框架

深度学习技术在过去十年中彻底改变了人工智能领域,成为语音识别、图像分析和自然语言处理的核心。选择合适的深度学习框架对开发者至关重要,影响项目的成功与可持续性。本文比较了主要框架(如TensorFlow、PyTorch和JAX),分析其技术基础、开发者体验及适用场景,帮助开发者根据项目需求做出最佳选择。

从开发者的角度比较深度学习框架

DEV Community
DEV Community · 2025-04-28T08:09:50Z

本文分析了Hugging Face的开源机器人库LeRobot中的π0模型,重点讨论了策略实现、配置系统、核心模型及注意力机制优化。通过对比JAX与PyTorch的实现,强调了两者在集成框架和适配性上的差异,并展示了如何将视觉-语言模型与专家模型结合,以生成精确的机器人动作序列。

LeRobot DP——LeRobot对动作策略Diffusion Policy的封装与解读(含DexCap库中对diffusion_policy的封装与实现)

结构之法 算法之道
结构之法 算法之道 · 2025-03-16T16:15:54Z
MPAX:基于JAX的新型线性规划工具,性能优于传统求解器,具备硬件加速

MPAX是一个基于JAX的开源线性规划工具,结合机器学习,采用先进的一阶方法,具备硬件加速和并行处理能力,性能优于传统求解器。

MPAX:基于JAX的新型线性规划工具,性能优于传统求解器,具备硬件加速

DEV Community
DEV Community · 2025-02-08T06:54:43Z

该研究提出Kozax框架,旨在解决遗传编程中适应性评估的高计算需求问题,支持大规模数据集和自定义运算符,展示了在科学计算中的优化潜力。

Kozax: Flexible and Scalable Genetic Programming in JAX

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2025-02-05T00:00:00Z
自主设计实验,速度提高2.5倍!马普所AI方法拓展显微镜使用边界

德国马克斯普朗克光科学研究所的研究团队开发了开源计算框架XLuminA,利用JAX加速光学显微镜实验设计,显著提升计算速度,支持新光学配置的自动发现,推动超分辨率显微镜技术的发展。该研究将于2024年12月10日发表在《Nature Communications》。

自主设计实验,速度提高2.5倍!马普所AI方法拓展显微镜使用边界

机器之心
机器之心 · 2024-12-13T07:41:56Z

本文介绍了Rust GPU项目及其在矩阵乘法中的应用,强调了Rust与GPU集成的优势。通过wgpu库实现了三种矩阵乘法kernel,展示了Rust在高性能计算中的潜力。同时,讨论了AWS Lambda与Rust结合提升登录性能的案例,以及molpipx库在化学领域的应用。

【Rust日报】2024-11-27 molpipx - JAX的Rust实现

Rust.cc
Rust.cc · 2024-12-01T01:54:36Z
Keras与JAX的比较

Keras和JAX是深度学习框架。Keras简化了TensorFlow,适合初学者;JAX专注于高性能计算,灵活性强,适合经验丰富的开发者。两者均支持GPU加速和自动微分,但Keras易用性更高,JAX则提供更高级的功能和优化。选择取决于用户需求。

Keras与JAX的比较

KDnuggets
KDnuggets · 2024-10-23T14:00:11Z

自动微分是一种高效计算函数导数的方法,结合了符号和数值微分的优点。它通过构建有向无环图并使用链式法则计算梯度,广泛应用于深度学习库如TensorFlow和PyTorch。JAX是支持自动微分的高性能库,适用于机器学习和科学计算。本文探讨了自动微分在数值相对论中的应用,展示了如何用JAX计算度量张量的导数。

使用JAX连接数值相对论与自动微分

DEV Community
DEV Community · 2024-10-19T00:06:33Z

深度学习在计算机视觉和自然语言处理中应用广泛。JAX由谷歌开发,支持自动微分和硬件加速,适合高性能项目。PyTorch由Facebook开发,易用且生态系统成熟,适合需要多样化库支持的项目。选择框架需根据项目需求。

JAX与PyTorch:机器学习框架中的两大强者对比

DEV Community
DEV Community · 2024-10-11T20:55:56Z
微调大模型,AMD MI300X就够了!跟着这篇博客微调Llama 3.1 405B,效果媲美H100

随着AI模型参数增加,算力需求也在增长。Felafax公司通过简化AI训练集群,将训练成本降低了30%。他们使用JAX在AMD GPU上微调LLaMA 3.1 405B模型,展示了JAX在非英伟达硬件上的优势。JAX支持多硬件并行,适应性强,迁移方便。Felafax利用JAX的设备网格功能进行参数分片,优化内存和计算效率,并通过LoRA技术减少可训练参数,实现高效微调。相关代码已开源,并提供详细教程。

微调大模型,AMD MI300X就够了!跟着这篇博客微调Llama 3.1 405B,效果媲美H100

机器之心
机器之心 · 2024-10-08T06:26:45Z

本研究提出了一种新算法,利用元胞自动机模型优化时间序列中的计算。通过预选有前景的规则,解决了规则选择问题。实验显示,该算法误差低,计算复杂度小,显著减少训练和优化时间。

基于JAX的细胞自动机加速库CAX

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2024-10-03T00:00:00Z

该研究开发了一种新算法结构,用于加速和可微分计算广义傅立叶变换,包括球谐变换和Wigner变换。算法在JAX框架中实现,结合自动和手动微分方法计算梯度,支持不同的球面采样,具有高精度和可扩展性,实现线性时间复杂度。

球面上的截断核随机梯度下降

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2024-10-02T00:00:00Z

本文介绍了脉冲神经网络(SNNs)在信号处理应用中的应用,讨论了算法和优化创新的最新进展,以及算法-架构共设计方面的研究成果。同时,描述了为有效利用这种算法创新而开发的基础硬件。最后,讨论了构建可部署SNN系统的研究前景和关键挑战。

SNNAX -- JAX中的脉冲神经网络

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2024-09-04T00:00:00Z

本研究在Jax中实现PCGRL环境,解决了现有程序内容生成(PCGRL)的规模限制问题,加快了训练速度。研究表明,模型能够学习到更稳健的设计策略,并在超出训练数据的地图尺寸上表现出良好的泛化能力,为设计师提供更大的控制权。

PCGRL+: 强化学习关卡生成的规模、控制与泛化

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2024-08-22T00:00:00Z

介绍了基于Python的JAX-Fluids求解器的第二个版本,用于可压缩的单相和两相流。新版本提供了增强的模拟能力和高阶数值离散方案。通过模拟验证了新增的数值模型。

JAX-SPH:一个可微分的平滑粒子流体动力学框架

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2024-03-07T00:00:00Z

介绍了Minigrid和Miniworld库,提供了目标导向的2D和3D环境,用于快速开发适用于各种研究需求的新环境。两个库被RL社区广泛采用,促进了广泛领域内的研究。通过案例研究展示了Minigrid和Miniworld统一API带来的额外功能,介绍了它们的设计哲学、环境细节和世界生成API。

XLand-MiniGrid:基于 JAX 的可扩展元强化学习环境

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2023-12-19T00:00:00Z

本文介绍了SMART平台,一个用于多机器人强化学习的仿真平台,提供多样化的交互场景进行训练,并支持基于插件的算法实现。研究者还开源了仿真环境、基准测试任务和基线模型,以推动多机器人强化学习的研究。

JaxMARL:基于 JAX 的多智能体强化学习环境

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2023-11-16T00:00:00Z

本研究开发了名为JAX-LOB的金融交易所限价挂单簿动态模拟器,能够高效模拟大规模交易。通过GPU加速,每条消息处理时间显著减少。研究还展示了如何通过强化学习解决最优执行问题,并分享了在GPU上进行端到端强化学习训练的初步结果。

JAX-LOB: 用于交易的大规模强化学习的 GPU 加速限价委托簿模拟器

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2023-08-25T00:00:00Z
  • <<
  • <
  • 1 (current)
  • 2
  • >
  • >>
👤 个人中心
在公众号发送验证码完成验证
登录验证
在本设备完成一次验证即可继续使用

完成下面两步后,将自动完成登录并继续当前操作。

1 关注公众号
小红花技术领袖公众号二维码
小红花技术领袖
如果当前 App 无法识别二维码,请在微信搜索并关注该公众号
2 发送验证码
在公众号对话中发送下面 4 位验证码
友情链接: MOGE.AI 九胧科技 模力方舟 Gitee AI 菜鸟教程 Remio.AI DeekSeek连连 53AI 神龙海外代理IP IPIPGO全球代理IP 东波哥的博客 匡优考试在线考试系统 开源服务指南 蓝莺IM Solo 独立开发者社区 AI酷站导航 极客Fun 我爱水煮鱼 周报生成器 He3.app 简单简历 白鲸出海 T沙龙 职友集 TechParty 蟒周刊 Best AI Music Generator

小红花技术领袖俱乐部
小红花·文摘:汇聚分发优质内容
小红花技术领袖俱乐部
Copyright © 2021-
粤ICP备2022094092号-1
公众号 小红花技术领袖俱乐部公众号二维码
视频号 小红花技术领袖俱乐部视频号二维码