Project Icon

torchsde

提供GPU支持的随机微分方程求解器

这个库提供了支持GPU和高效反向传播的随机微分方程(SDE)求解器。其使用Python和PyTorch开发,安装方便,并附有丰富的示例。用户可以通过简单的代码示例快速入门,并通过文档进一步学习。除了基础功能外,还包括潜在SDE和GAN中的SDE等高级应用示例。适用于在高性能计算环境中执行复杂SDE模型的研究人员和开发者。

DeepSpeed - 一个深度学习优化库,专为大规模模型训练和推理设计
DeepSpeedGithub分布式训练大规模模型训练开源项目模型压缩模型推理
DeepSpeed 是一个深度学习优化软件套件,专为大规模模型训练和推理设计,能显著优化系统性能和降低成本。它支持亿级至万亿级参数的高效管理,兼容各种计算环境,从资源受限的GPU系统到庞大的GPU集群。此外,DeepSpeed 在模型压缩和推理领域亦取得创新成就,提供极低的延迟和极高的处理速率。
autoregressive-diffusion-pytorch - 自回归扩散模型:无向量量化的图像生成方法
GithubPyTorch图像生成开源项目深度学习神经网络自回归扩散
autoregressive-diffusion-pytorch是一个基于PyTorch的自回归扩散模型实现,源自'Autoregressive Image Generation without Vector Quantization'论文。模型支持序列和图像输入,无需向量量化即可生成高质量图像。项目提供简洁API接口,包含详细使用说明和示例代码,适合研究人员和开发者探索自回归扩散模型。
KoopmanLab - Koopman神经算子 高效求解非线性偏微分方程
GithubKoopmanLabKoopman神经算子偏微分方程开源项目机器学习物理方程求解
KoopmanLab是一个开源的Koopman神经算子包,基于PyTorch开发。该项目结合机器学习和动力系统理论,提供了一种无网格方法来求解非线性偏微分方程。KoopmanLab实现了多种模型,如KNO和ViT-KNO,并配备完整的数据处理、训练和测试工具。它可应用于Navier-Stokes方程和浅水方程等物理模拟场景,为研究人员提供了高效灵活的计算框架。
image-restoration-sde - 创新图像恢复方法 结合SDE和扩散模型的IR-SDE与Refusion
GithubIR-SDERefusionSDE图像恢复开源项目深度学习
该项目提出IR-SDE和Refusion两种图像恢复方法。IR-SDE采用均值回复随机微分方程,在多项任务中达到最优性能。Refusion整合潜空间扩散模型,可处理大尺寸真实图像。这些技术适用于合成和实际数据集,有效解决图像去雨、去雾、去阴影等问题。项目开源完整PyTorch实现代码,并提供预训练模型和使用指南。
pyg-lib - 图形神经网络高性能计算库
CUDAGithubPyTorchPythonpyg-lib安装开源项目
pyg-lib是一款专为图形神经网络优化的高性能计算库。该项目为Linux、Windows和macOS等主流操作系统提供预构建的Python包,兼容多个PyTorch版本和CUDA组合,支持Python 3.8至3.12。pyg-lib通过提升图形神经网络的计算效率,为研究和开发提供了实用工具。研究人员和开发者可根据具体的系统环境,使用pip命令快速安装所需版本,轻松增强图形神经网络的性能。
ETSformer-pytorch - 基于PyTorch的先进时间序列Transformer模型
ETSformerGithubPytorchTransformer开源项目指数平滑时间序列预测
ETSformer-pytorch是一个开源的时间序列分析工具,基于PyTorch实现了先进的Transformer模型。该项目集成了多头指数平滑注意力机制和频率选择功能,适用于时间序列预测和分类任务。ETSformer-pytorch提供简单的安装和使用方法,支持灵活的模型配置,并包含专门的分类包装器。这一工具为研究人员和开发者提供了处理复杂时间序列数据的有效解决方案。
lightning-bolts - PyTorch Lightning的深度学习组件扩展
BoltsDeepSparseGithubPyTorch LightningSparseMLTorch ORT开源项目
Lightning Bolts为PyTorch Lightning提供了多种扩展组件,包括回调和数据集,旨在加速训练和推理。它支持通过Torch ORT将模型转换为优化的ONNX图,以实现GPU加速训练;并通过SparseML在微调中引入稀疏性,提高推理性能。项目支持广泛的问题解决,并欢迎用户贡献通用组件。了解更多安装和使用信息,请访问官方文档和社区支持平台。
PPO-PyTorch - 使用PyTorch实现的简易PPO算法工具
GithubOpenAI gymPPO-PyTorchProximal Policy Optimization开源项目强化学习超参数调整
该项目提供了一个基于PyTorch的简易PPO算法实现,适用于OpenAI Gym环境,帮助初学者理解PPO。更新内容包括整合离散和连续算法,并引入了线性衰减机制。用户可以通过PPO_colab.ipynb在Google Colab中便捷地训练、测试和绘制图表。项目支持单线程执行,并提供日志记录、可视化和动画生成工具。
pytorch-fid - 生成对抗网络图像质量评估工具
FIDFréchet Inception DistanceGANsGithubPyTorchTensorflow开源项目
pytorch-fid是一款用于计算生成对抗网络(GAN)样本质量的Fréchet Inception Distance(FID)分数的工具。该工具将官方的Tensorflow实现移植到PyTorch,确保相似的准确性和方便性。用户可以自由选择特征层,适应不同的数据集,还支持GPU加速和保存原始数据集的统计信息,便于进行多模型比较,适合研究和开发高质量图像生成模型。
autograd - 支持Python和NumPy的高效自动微分库
AutogradGithubNumPyPython开源项目梯度优化自动微分
Autograd是一个自动微分库,可对原生Python和NumPy代码进行微分。它支持反向模式和前向模式微分,能高效计算标量函数对数组参数的梯度。Autograd兼容Python的多数特性,如循环、条件语句、递归和闭包,并支持高阶导数。这个库主要应用于基于梯度的优化,在机器学习、神经网络和科学计算等领域有广泛应用。
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

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

Project Cover

AI写歌

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

Project Cover

有言AI

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

Project Cover

Kimi

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

Project Cover

阿里绘蛙

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

Project Cover

吐司

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

Project Cover

SubCat字幕猫

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

Project Cover

美间AI

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

Project Cover

稿定AI

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

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号