Reddit · MachineLearning· /u/Important-Home4431·· 4 小时前AI 评分35
Deepity:手调 C++/CUDA 预测编码库在 MNIST CPU 基准测试中超越 jpc 与 ngc-learn
Deepity: a hand-tuned C++/CUDA predictive coding library, beats jpc (JAX) and ngc-learn on MNIST (CPU benchmarks + source inside) [P]
AI 摘要
Deepity 发布首个公开版本,在 MNIST 数据集上以 98.21% 准确率耗时 487.5 秒,优于 jpc (97.55%/531.4s) 和 ngc-learn (98%/1707.7s)。该库基于 C++20 核心与手写 AVX2/AVX-512 SIMD 内核,提供 Python 绑定及多种 PCN 变体。目前为 0.1.0a3 版,CPU 路径已成熟,CUDA 后端仍在优化中。
来源:Reddit · MachineLearning · reddit.com