Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.


tutorials Documentation Status PyPI version PyPI downloads Docker Pulls license

PERSIA (Parallel rEcommendation tRaining System with hybrId Acceleration) is developed by AI platform@Kuaishou Technology, collaborating with ETH. It is a PyTorch-based (the first public one to our best knowledge) system for training large scale deep learning recommendation models on commodity hardwares. It is capable of training recommendation models with up to 100 trillion parameters. To the best of our knowledge, this is the largest model size in recommendation systems so far. Empirical study on public datasets indicate PERSIA's significant advantage over several other existing training systems in recommendation [1]. Its efficiency and robustness have also been validated by multiple applications with 100 million level DAU at Kuaishou.

Disclaimer: The program is usable and has served several important businesses. However, the official English documentation and tutorials are still under heavy construction, but they are a bit raw now). We encourage adventurers to try out PERSIA and contribute!

News

  • AI Engines in the "Short-video" Era: Eating 100 Trillion Parameters, Invited talk, Facebook, 2021.
  • 单机训练速度提升 640 倍!独家解读快手商业广告模型 GPU 训练平台 PERSIA (In Chinese. Title: 640x Faster GPU Based Learning System for Ad Recommendation)
  • 创新、平衡与大格局:快手商业化的慢与快 (In Chinese. Title: Innovation, Balance, and Big Picture: The Speed of Kwai Commercialization)

Discussion

Feel free to join our Telegram Group for discussion!

References

  1. Xiangru Lian, Binhang Yuan, Xuefeng Zhu, Yulong Wang, Yongjun He, Honghuan Wu, Lei Sun, Haodong Lyu, Chengjun Liu, Xing Dong, Yiqiao Liao, Mingnan Luo, Congfei Zhang, Jingru Xie, Haonan Li, Lei Chen, Renjie Huang, Jianying Lin, Chengchun Shu, Xuezhong Qiu, Zhishan Liu, Dongying Kong, Lei Yuan, Hai Yu, Sen Yang, Ce Zhang, & Ji Liu. (2021). Persia: A Hybrid System Scaling Deep Learning Based Recommenders up to 100 Trillion Parameters.

  2. Ji Liu & Ce Zhang. (2021). Distributed Learning Systems with First-order Methods.

License

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Metadata

Release files for persia 0.1.dev234

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for persia 0.1.dev234
File Size Uploaded Provenance
persia-0.1.dev234.tar.gz 211.0 kB Without provenance Details

Built distributions (wheels)

Table of built distributions (wheels) for persia 0.1.dev234
File
persia-0.1.dev234-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.9 CPython 3.9 Linux glibc 2.17+ x86-64 Details
persia-0.1.dev234-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.8 CPython 3.8 Linux glibc 2.17+ x86-64 Details
persia-0.1.dev234-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.7 CPython 3.7 pymalloc Linux glibc 2.17+ x86-64 Details
persia-0.1.dev234-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.6 CPython 3.6 pymalloc Linux glibc 2.17+ x86-64 Details

Total release size: 153.5 MB

Release files / persia-0.1.dev234.tar.gz

Download URL persia-0.1.dev234.tar.gz
Size 211.0 kB
Tags Source
SHA-256 checksum
How to use checksums
6a74c8f3f7a56f8a5a222927ed8cf9e3a62d20308738d3774957684ca94702db
BLAKE2b-256 checksum
How to use checksums
e318a7bc26642a53c83c497e9977f7ac13ddc67f1ff85ffb16cbb9d5b68900d1
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

Release files / persia-0.1.dev234-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL persia-0.1.dev234-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 38.3 MB
Tags CPython 3.9 Linux glibc 2.17+ x86-64
SHA-256 checksum
How to use checksums
8cb64849d1ef22c918d57ae38628477323b0fd36cd898de5ec014fcdd642347c
BLAKE2b-256 checksum
How to use checksums
288c8b183cc55ebdc8036d443e7ec8dcaddc95689c9e383c674d2b37ceeaa7bf
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

Release files / persia-0.1.dev234-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL persia-0.1.dev234-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 38.3 MB
Tags CPython 3.8 Linux glibc 2.17+ x86-64
SHA-256 checksum
How to use checksums
01e7b9590545d3a941b92e681c6b69006a459fd2e5bf1a74ce50289d70fe3cf7
BLAKE2b-256 checksum
How to use checksums
c157b3235f3e0181ba732acd38e94f55bd29a9939da319c2080d4c3c7007a2d8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

Release files / persia-0.1.dev234-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL persia-0.1.dev234-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 38.3 MB
Tags CPython 3.7 CPython 3.7 pymalloc Linux glibc 2.17+ x86-64
SHA-256 checksum
How to use checksums
f4b5158a544872a5e58a2c9d4b50d6430bb9eea6adcb5bd3e80080bf238a2521
BLAKE2b-256 checksum
How to use checksums
73d121d99bdd1a80bc5ce6c6ccaefac3bbda3d8d965c6700f8d708b7ffa42aa5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

Release files / persia-0.1.dev234-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL persia-0.1.dev234-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 38.3 MB
Tags CPython 3.6 CPython 3.6 pymalloc Linux glibc 2.17+ x86-64
SHA-256 checksum
How to use checksums
84220eaf0ba1d684de5438c0fa912a956fdb6dafeee774a784b8db9c7de62c96
BLAKE2b-256 checksum
How to use checksums
a02c55aca65fedc7d153748bc44cc6e3e2d75db40021854e7cb18b739f0278db
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

Security for persia 0.1.dev234

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

Release history Release notifications | RSS feed

This release
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