Skip to main content

🔥 Pyro 框架

轻量级深度学习框架,从零手写,只依赖 NumPy。

特点

  • 纯 Python + NumPy,无需 GPU
  • 自动求导
  • 内置 Transformer 组件
  • 专为移动端设计

安装

pip install -e .
快速开始

```python
from pyro import Tensor, Linear, Adam

x = Tensor([1, 2, 3], requires_grad=True)
loss = (x * x).sum()
loss.backward()

模块

· Tensor:自动求导张量 · Linear:全连接层 · Attention:多头注意力 · Loss:交叉熵/均方误差 · Optim:SGD/Adam EOF

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pyro_ai-0.1.0-py3-none-any.whl (10.6 kB view details)

Uploaded Python 3

File details

Details for the file pyro_ai-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: pyro_ai-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 10.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.13.13

File hashes

Hashes for pyro_ai-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 990ab0af10ed9d803de07748610e11bfa265d232ab706c1218a8b7bb7d3d8363
MD5 95e4d32593a856b210d06c8e541dfe8e
BLAKE2b-256 8835f8b567582bcd713694eb9189b07a9d0b67f7c64182117dfed4ba01f5f7d1

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

0.1.0 This release

1 file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page