sbx

Jax加持的Stable-Baselines3强化学习库

Stable Baselines Jax 强化学习机器学习算法实现开源项目 Github

SBX是Stable-Baselines3的Jax实现版本，集成了SAC、TQC、PPO等多种先进强化学习算法。它与SB3保持相同API，可与RL Zoo无缝对接，并提供详细使用示例。SBX为复杂环境和任务提供高效、可靠的强化学习实现。

文档

DRL-Pytorch - PyTorch实现的深度强化学习算法集合

DRL算法GithubPyTorch人工智能开源项目强化学习深度学习

DRL-Pytorch项目提供多种常用深度强化学习算法的PyTorch实现,包括Q-learning、DQN变体、PPO、DDPG、TD3和SAC等。代码结构清晰统一,便于研究人员和开发者比较不同算法。项目还包含详细使用说明、依赖列表和学习资源推荐,有助于快速入门和实践。

rl_games - 强化学习框架支持多环境及算法的高性能实现

GPU加速GithubRL Games多智能体训练开源项目强化学习机器人学习

rl_games是一个高性能强化学习库，实现了PPO、A2C等算法，支持NVIDIA Isaac Gym、Brax等环境的GPU加速训练。该库具备异步actor-critic、多智能体训练、自对弈等功能，可在多GPU上并行。rl_games提供Colab notebook示例便于快速上手，在多个基准测试中表现出色。作为一个功能丰富的强化学习工具，rl_games兼具高性能和易用性。

DeepRL - PyTorch 中深度强化学习算法的模块化实现

A2CDQNDeepRLGithubPyTorch开源项目深度强化学习

DeepRL项目使用PyTorch实现了一系列流行的深度强化学习算法，提供模块化框架，适用于从简单任务到高难度游戏。支持的算法包括DQN、C51、QR-DQN、A2C、DDPG、PPO等，并具备异步数据生成和传输功能。项目依赖PyTorch v1.5.1，具体依赖请参考Dockerfile和requirements.txt。此外，项目提供代码示例和性能曲线图，适合相关研究参考和使用。

nanodl - 设计与训练变压器模型的Jax库

GithubJaxNanoDLtransformer模型分布式训练开源项目深度学习

这是一个基于Jax的库，旨在简化变压器模型的开发和训练，特别适合资源有限的环境。支持多种模型如Gemma、GPT3、T5和Whisper，涵盖自然语言处理和计算机视觉任务。提供灵活的模块和层，包括Jax/Flax中未提供的RoPE、GQA、MQA和Swin注意力机制，支持多GPU/TPU的数据并行训练，简化数据处理。该库还包含加速的经典机器学习模型，帮助用户以最小的代码重写快速实现模型开发和训练。

Safe-Policy-Optimization - 安全强化学习的全面算法基准平台

GithubPKU-AlignmentSafe-Policy-OptimizationSafety-Gymnasium安全强化学习开源项目算法基准

Safe-Policy-Optimization为安全强化学习(Safe RL)提供了全面的算法基准平台。该项目整合了多种算法和环境，支持单智能体和多智能体任务，具备正确性、可扩展性、日志记录和可视化等特性。通过统一的接口和详细文档，Safe-Policy-Optimization简化了安全RL算法的评估和比较流程，为研究人员提供了强大的实验工具。

rsl_rl - 面向GPU的高效强化学习框架

GPU运行GithubPPO算法RSL RL开源项目强化学习

rsl_rl是一个专为GPU运行优化的强化学习框架，目前实现了PPO算法，未来将支持更多算法。框架提供详细的安装指南，集成多种日志工具，并采用严格的代码质量管理。它在Legged-Gym和Orbit等机器人仿真环境中得到应用，为强化学习研究和开发提供了高效工具。

morl-baselines - 多目标强化学习算法库支持单策略和多策略实现

GithubMO-GymnasiumMORL-BaselinesPyTorch多目标强化学习开源项目算法库

MORL-Baselines是一个多目标强化学习算法库，提供多种PyTorch实现。该项目遵循MO-Gymnasium API，支持单策略和多策略算法，适用于SER和ESR标准。特点包括自动性能报告、代码规范和自动测试。实现了GPI-LS、MORL/D等多种算法，支持连续和离散观察/动作空间，为MORL研究和基准测试提供有力支持。

EasyLM - 简化的大规模语言模型训练与部署

EasyLMGPT-JGithubJAXLLaMATPU开源项目

EasyLM提供了一站式解决方案，用于在JAX/Flax中预训练、微调、评估和部署大规模语言模型。通过JAX的pjit功能，可以扩展到数百个TPU/GPU加速器。基于Hugginface的transformers和datasets，EasyLM代码库易于使用和定制。支持Google Cloud TPU Pods上的多TPU/GPU和多主机训练，兼容LLaMA系列模型。推荐加入非官方的Discord社区，了解更多关于Koala聊天机器人和OpenLLaMA的详细信息及安装指南。

big_vision - 基于Jax/Flax的大规模视觉模型训练框架

GithubJaxbig vision开源项目模型训练深度学习计算机视觉

Big Vision是一个用于训练大规模视觉模型的开源代码库。它基于Jax/Flax构建，支持在Cloud TPU VM和GPU上运行。该项目采用tf.data和TensorFlow Datasets实现高效的数据处理，可无缝扩展至2048个TPU核心的分布式环境。Big Vision涵盖了视觉Transformer、多模态学习、知识蒸馏等多个研究方向，为大规模视觉实验提供了可靠的基础。

scenic - 多模态视觉智能研究框架

GithubJAXScenicTransformer开源项目深度学习计算机视觉

Scenic是一个基于JAX的开源视觉智能研究框架,聚焦注意力机制模型。它提供轻量级共享库和完整项目实现,支持分类、分割、检测等任务,可处理图像、视频、音频等多模态数据。Scenic内置多个前沿模型和基线,有助于快速原型设计和大规模实验。

相关项目

推荐项目

豆包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