jax

高性能科学计算和机器学习的Python加速库

JAX是一个专为高性能数值计算和大规模机器学习设计的Python库。它利用XLA编译器实现加速器导向的数组计算和程序转换，支持自动微分、GPU和TPU加速。JAX提供jit、vmap和pmap等函数转换工具，让研究人员能够方便地表达复杂算法并获得出色性能，同时保持Python的灵活性。

访问官网

Github

文档

论文

介绍相关项目

torchquad - 基于GPU加速的开源数值积分框架

GPUGithubPyTorchtorchquad开源项目数值积分机器学习

torchquad是一个开源的高性能数值积分框架,支持PyTorch、JAX和Tensorflow等多个后端。该框架针对GPU进行了优化,能有效处理高维积分问题,并在GPU上展现出优异的扩展性。torchquad提供多种积分方法,支持自动微分,适用于机器学习和科学计算等领域。其简洁的API设计使研究人员和开发者能够高效地完成复杂的数值积分任务。

max - 一套集成的AI库、工具和技术

AIGithubMAX工具链开源项目推理硬件可移植性

MAX平台是一套集成的AI库、工具和技术，统一了分散的AI部署工作流。通过提供单一开发工具链，MAX显著缩短了创新产品的上市时间，同时具备完全编程能力、卓越的性能和顺畅的硬件兼容性。文档、代码示例和Jupyter笔记本等资源可帮助用户快速起步，并提供社区支持和交流。

EasyDeL - 多模型训练优化框架

EasyDeLFlaxGithubJAX开源项目机器学习模型训练

EasyDeL是一个开源框架，用于通过Jax/Flax优化机器学习模型的训练，特别适合在TPU/GPU上进行大规模部署。它支持多种模型架构和量化方法，包括Transformers、Mamba等，并提供高级训练器和API引擎。EasyDeL的架构完全可定制和透明，允许用户修改每个组件，并促进实验和社区驱动的开发。不论是前沿研究还是生产系统构建，EasyDeL都提供灵活强大的工具以满足不同需求。最新更新包括性能优化、KV缓存改进和新模型支持。

dinosaur - 全球大气建模的新工具可微分动力学谱系核心

DinosaurGithubJAX动力核心大气建模开源项目自动微分

Dinosaur是一个基于JAX的全球大气建模谱系动力核心。它可以求解浅水方程和原始方程，支持自动微分，并针对GPU/TPU等硬件优化。项目提供了标准测试案例，如正压不稳定性和Held-Suarez强迫。Dinosaur为大气科学研究提供了高效的并行计算工具，有助于推进大气动力学研究。

Flux.jl - 轻量级的纯Julia机器学习框架

FluxGPU支持GithubJulia开源项目机器学习轻量级抽象

Flux.jl是一个基于100%纯Julia技术栈的机器学习框架，结合原生GPU和自动微分支持，提供简洁优雅的解决方案。该框架使简单任务易于实现，并且高度可定制。适用于Julia 1.9及更高版本，支持快速模型构建与训练。了解更多信息，请参阅快速入门和文档页面。

deepxde - 科学计算与物理学习的深度学习库

DeepXDEGithub开源项目深度学习库物理信息学习神经网络科学机器学习

DeepXDE 是一个为科学计算和物理引导学习设计的深度学习库。它支持解决多种复杂问题，如常微分方程、偏微分方程、分数阶微分方程和随机微分方程等。DeepXDE 支持多个后端，包括 TensorFlow、PyTorch、JAX 和 PaddlePaddle，提供丰富的几何域、边界条件、自动微分和采样方法。其模块化设计允许用户自定义和扩展模块，适用于科研和工业应用。

trax - 代码清晰、高速执行的深度学习库

GithubGoogle BrainReformerTransformerTrax开源项目深度学习

Trax是一个由Google Brain团队维护的端到端深度学习库，专注于清晰代码和高速执行。它提供预训练的Transformer模型和丰富的API文档，支持用户创建和训练自定义模型，并与TensorFlow数据集无缝集成。Trax兼容CPUs、GPUs和TPUs，用户可以通过Python脚本、notebooks和命令行界面轻松使用。

axon - 基于Nx框架的Elixir深度学习库，支持模型创建和训练

AxonElixirGithub开源项目模型创建深度学习训练API

Axon是一个基于Nx的Elixir深度学习库，提供高效的神经网络功能。核心组件包括数值定义的Functional API、模型创建的Model Creation API和训练模型的Training API。Axon通过模块化设计，实现各API独立使用，并通过Polaris进行优化。该库支持多种深度学习层和模型格式，如TensorFlow Lite和ONNX，适用于多种深度学习应用。

transformers - 机器学习库，覆盖文本、视觉与音频处理

GithubHugging Face人工智能多模态开源项目机器学习自然语言处理

探索🤗 Transformers——一个功能全面的机器学习库，覆盖文本、视觉与音频处理。该库提供数千种可对接JAX、PyTorch或TensorFlow的预训练模型，适用于多种语言处理与多模态任务。主要功能包括： - 文本分类 - 信息提取 - 问答系统 - 摘要生成 - 翻译 - 文本生成此外，还能处理表格问答、OCR及视觉问答等多模态任务。Transformers库易于使用，支持模型间的快速切换与无缝整合。

apex - NVIDIA Apex加速PyTorch混合精度与分布式训练

ApexGithubNVIDIAPyTorch分布式训练开源项目混合精度训练

Apex是NVIDIA开发的PyTorch扩展库，专注于优化混合精度和分布式训练。该工具提供自动混合精度、分布式数据并行和同步批量归一化等功能，大幅提高训练效率。Apex还集成了多个CUDA优化扩展，如快速层归一化和融合优化器，进一步增强性能。作为持续更新的开源项目，Apex为PyTorch用户提供了最新的训练加速工具。

相关项目

推荐项目

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手，通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能，支持100+编程语言，与主流编辑器无缝集成，显著提升开发效率和代码质量。

AI写歌

Suno AI是一个革命性的AI音乐创作平台，能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐，Suno AI都是音乐爱好者和专业人士的理想选择。

有言AI

有言平台提供一站式AIGC视频创作解决方案，通过智能技术简化视频制作流程。无论是企业宣传还是个人分享，有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Kimi

Kimi AI助手提供多语言对话支持，能够阅读和理解用户上传的文件内容，解析网页信息，并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题，Kimi都能以友好、专业的方式提供帮助。

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术，为商家提供一键生成商品图和营销文案的服务，显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台，让商品第一时间被种草。

吐司

探索Tensor.Art平台的独特AI模型，免费访问各种图像生成与AI训练工具，从Stable Diffusion等基础模型开始，轻松实现创新图像生成。体验前沿的AI技术，推动个人和企业的创新发展。

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器，它将改变您观看视频的方式！SubCat结合了先进的人工智能技术，为您提供即时视频字幕翻译，无论是本地视频还是网络流媒体，让您轻松享受各种语言的内容。

美间AI

美间AI创意设计平台，利用前沿AI技术，为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图，再到文案生成，美间让创意设计更简单、更高效。

稿定AI

稿定设计是一个多功能的在线设计和创意平台，提供广泛的设计工具和资源，以满足不同用户的需求。从专业的图形设计师到普通用户，无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑，稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合，帮助用户轻松实现创意设计。

使用协议隐私政策广告服务

投诉举报邮箱: service@vectorlightyear.com