跳到正文
Hacker News · Show HN· paramthakkar·· 3 小时前精选AI 评分75

Simulusion:一个用于训练和运行世界模型的 PyTorch 库

Show HN: Simulusion – A PyTorch library for training and running world models

AI 导读

Simulusion 是一个开源的 PyTorch 库,统一了梦幻、PlaNet、JEPA 等世界模型的创建、配置和训练接口,支持自定义编码器、解码器与潜在动力学骨干,并集成 Gym、DeepMind Control 等环境,提供 ONNX、TorchScript、TensorRT 导出选项。

推荐理由

适合希望快速对比梦幻、PlaNet等世界模型的研究人员和开发者,pip安装后在单张RTX 3050上两小时内即可训练出演示模型,提供统一API、环境集成和可视化工具,降低入门门槛。

来源:Hacker News · Show HN · github.com