Wheelodex » Projects » deepspeed

deepspeed

View on PyPI — Reverse Dependencies (86)

0.3.1.dev5 deepspeed-0.3.1.dev5-py3-none-any.whl

Wheel Details

Project: deepspeed
Version: 0.3.1.dev5
Filename: deepspeed-0.3.1.dev5-py3-none-any.whl
Download: [link]
Size: 247609
MD5: 960672995166b28e92a4183f3a970ef3
SHA256: 8077d5af79f3e1e4d1cef0a789d17390ead1ce2d21e875894455529a26af158a
Uploaded: 2020-11-11 22:22:32 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: deepspeed
Version: 0.3.1.dev5
Summary: DeepSpeed library
Author: DeepSpeed Team
Author-Email: deepspeed[at]microsoft.com
Home-Page: http://deepspeed.ai
License: MIT
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Requires-Dist: torch (>=1.2)
Requires-Dist: torchvision (>=0.4.0)
Requires-Dist: tqdm
Requires-Dist: psutil
Requires-Dist: tensorboardX (==1.8)
Requires-Dist: ninja
Requires-Dist: cpufeature
Requires-Dist: mpi4py; extra == "1bit_adam"
Requires-Dist: cupy-cuda100; extra == "1bit_adam"
Requires-Dist: clang-format; extra == "all"
Requires-Dist: sphinx-rtd-theme; extra == "all"
Requires-Dist: pytest; extra == "all"
Requires-Dist: recommonmark; extra == "all"
Requires-Dist: cupy-cuda100; extra == "all"
Requires-Dist: psutil; extra == "all"
Requires-Dist: tqdm; extra == "all"
Requires-Dist: pre-commit; extra == "all"
Requires-Dist: mpi4py; extra == "all"
Requires-Dist: pytest-forked; extra == "all"
Requires-Dist: sphinx; extra == "all"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-forked; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: clang-format; extra == "dev"
Requires-Dist: sphinx; extra == "dev"
Requires-Dist: recommonmark; extra == "dev"
Requires-Dist: sphinx-rtd-theme; extra == "dev"
Requires-Dist: tqdm; extra == "readthedocs"
Requires-Dist: psutil; extra == "readthedocs"
Provides-Extra: 1bit_adam
Provides-Extra: all
Provides-Extra: dev
Provides-Extra: readthedocs
[No description]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.35.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
deepspeed/__init__.py sha256=SUOQOKG8WAD4USABA6K_rdAramYEU_LGvsHITsCkI7Y 7678
deepspeed/env_report.py sha256=PVu5rvIm9ag_6AIBm7P2kzFPnym5Im25DqX9ZuwQrHA 3505
deepspeed/git_version_info.py sha256=mQiGn-5ytHplhKQ_bMfj1kwX41a8kCOHYAj85IOkAyY 520
deepspeed/git_version_info_installed.py sha256=XPyci8QNrH1hvF7u356L3GFq4NOdJLWrcqgG6_s6bsU 465
deepspeed/launcher/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
deepspeed/launcher/constants.py sha256=uPAGsnmc50fRRAiXaaVxcHeXu0ZOobs4NmpANTWGQFs 387
deepspeed/launcher/launch.py sha256=WBOiyO99OmSRhYAT6cMhEXjORR_dxnsBTXDH4oaE1Vs 4886
deepspeed/launcher/multinode_runner.py sha256=7HYOq9kJVnlza_Wakp-aUmjVBTR2at-gJEs3m7p-MGI 6912
deepspeed/launcher/runner.py sha256=uyU8UXWoarZgvSAp84LFoyoLufCPSOlznfH6vKEaQcM 13766
deepspeed/ops/__init__.py sha256=4LDirsnhbi4axQI_6Nm-3-3Psa3XNFiMSS0Ob8r4DK4 164
deepspeed/ops/adam/__init__.py sha256=J4JaBxZ-6bKoVpR_GhVnBp8xzUI2zi2iDvlMxpYQzjs 75
deepspeed/ops/adam/cpu_adam.py sha256=e5EBuTm2yMoOSfXHxW7WZylZ3yhfq4x-8gomIy72L3Q 5989
deepspeed/ops/adam/fused_adam.py sha256=22OrNccpZbUt0pMC7h9favjFro0H-zB8vD7DXlEu0xc 7445
deepspeed/ops/adam/multi_tensor_apply.py sha256=vTH6_pDXRB2ZN_aflb6aETf1-7ZttF3Wa9JAPumVI4Y 391
deepspeed/ops/csrc/adam/compat.h sha256=0nQ1ftyORrphhZbhmr87BklaCdHc3pf_16BaTxB_iNQ 287
deepspeed/ops/csrc/adam/cpu_adam.cpp sha256=s--t-f8JtxQvKV6Vv55zvIQH5eKk8Jdgo4D6cx7O-MQ 33698
deepspeed/ops/csrc/adam/custom_cuda_kernel.cu sha256=lSk3NvK6sz_6xv0yiGKWLb6AVTjONFseRlhHPqKGN9g 1232
deepspeed/ops/csrc/adam/fused_adam_frontend.cpp sha256=2VMubaicYG_bA3a1JIuOEm2f7Ej5ytutggL08bdXo2c 781
deepspeed/ops/csrc/adam/multi_tensor_adam.cu sha256=yKdpk5yVh7u8s5kJqpuFcKdF7heolBKc7FijJhtBDVc 6524
deepspeed/ops/csrc/adam/multi_tensor_apply.cuh sha256=3uqapm339T7GnxzVxwr3_j81hmUzJTA59aELhSL_d04 5546
deepspeed/ops/csrc/includes/StopWatch.h sha256=CO9Jd9D6pDSPWZtLhHuOCsxA5ID5OmxgPIfmJjxiyXw 1882
deepspeed/ops/csrc/includes/Timer.h sha256=JvhS5dTJpreLUphjtwAVvP9pq2ZGMYd2nc4nRdDc94I 1129
deepspeed/ops/csrc/includes/context.h sha256=9SL8qEzsQ_QoEY08bk1AJc4oBc08g0UDdKlutmHdRPg 6934
deepspeed/ops/csrc/includes/cpu_adam.h sha256=MbYH7CwWUOxbTog3E0smGHqvblp7M69dAKdHwDiEDKc 4050
deepspeed/ops/csrc/includes/cublas_wrappers.h sha256=EZscjH2jVqFoYQqiW_okvOqyq6P42SfC5Rz6ZPsWzvc 2660
deepspeed/ops/csrc/includes/custom_cuda_layers.h sha256=yZjIUKKYUnfXbrArfEK_LXcAfJBhY5qs1kYIOcIuDzI 10093
deepspeed/ops/csrc/includes/dropout.h sha256=7nY-MEfuvnUbcxWzquR0sjvGqYf52FZSEV_BbJiQJNY 2172
deepspeed/ops/csrc/includes/ds_transformer_cuda.h sha256=lMPyXaqssDfnaYHFwd40sBSddMiSrIXVXZTYeHjlTaY 5718
deepspeed/ops/csrc/includes/feed_forward.h sha256=rIEwVOzdbB1084ams0sKkw3HyRcguLIYudiDRSIZHvI 2862
deepspeed/ops/csrc/includes/gelu.h sha256=RUw1IKrstEJfPUdSNhHvKVs7KXAN0lZX88Kq57WsFG4 955
deepspeed/ops/csrc/includes/gemm_test.h sha256=1eluUEbouNaiketQFRQiM4L_lxWJ-5lMrXoUoWvTqIc 9526
deepspeed/ops/csrc/includes/general_kernels.h sha256=tNGdNwSHIidJYbA3dAO1_YvF03G5W4O1lFWQ2M51Ui4 1374
deepspeed/ops/csrc/includes/normalize_layer.h sha256=g_FPVMUGnTpSURlHVTYGxb0xMZXx_KL23VAPVSgGVWU 7080
deepspeed/ops/csrc/includes/softmax.h sha256=fVtkUFU776X-iPnG90vukZB9Q194TfyQ8G_JZVtMVCs 1601
deepspeed/ops/csrc/includes/strided_batch_gemm.h sha256=XHiGAq_F7Zhl86X3vKZAT2usVCP07P3Aab_Oc0Ctv_s 6407
deepspeed/ops/csrc/includes/type_shim.h sha256=oqBg0fAdxk2fAxcZh11EPXFUoc13BsTj2ONs6j0zvPs 5458
deepspeed/ops/csrc/lamb/fused_lamb_cuda.cpp sha256=2CT5wh0rydobJQOEdJkjyUvTkd3TGAc03zurDkldLqw 3953
deepspeed/ops/csrc/lamb/fused_lamb_cuda_kernel.cu sha256=HwGwXfNfG8CKMbnxGeJXKzjpRjAoCiqeXivUsVNACG8 15080
deepspeed/ops/csrc/sparse_attention/utils.cpp sha256=elxj0dDcpgOchQNzISoxeJNu2QGKSO5KlVsIk4dbw1c 4419
deepspeed/ops/csrc/transformer/cublas_wrappers.cu sha256=roKqoimKJgyc_kvndJubqyMYCr0FV1vbKheeXtj4uoA 8559
deepspeed/ops/csrc/transformer/dropout_kernels.cu sha256=bGfVQVj4f-vOpb6Br-mqrndthw7iUklnPFPNn-uMgyk 24493
deepspeed/ops/csrc/transformer/ds_transformer_cuda.cpp sha256=R9_bQR3qRJLjq3F7LsoMysZPuMDrA_Fac79HsTNf3Xw 45760
deepspeed/ops/csrc/transformer/gelu_kernels.cu sha256=68RypYnjZLJDhYjF1YKg9PFGGmMBpzLimiECwKBTFuQ 12727
deepspeed/ops/csrc/transformer/general_kernels.cu sha256=M1OMi8QKyhaP2__v3dT2m9jafX5nNypOHqSrJDFXXwU 15047
deepspeed/ops/csrc/transformer/normalize_kernels.cu sha256=2Bi8u5kkDRsGNeUvahjP0QUfDNWdBDZnIKF27yfJrV4 79696
deepspeed/ops/csrc/transformer/softmax_kernels.cu sha256=z9AZQnRnLAaIzhgirE5kWHODOLooKxRlXAmfgmwDH68 22610
deepspeed/ops/csrc/transformer/transform_kernels.cu sha256=diC-Kiw9bItUaXegXbyFdR0sPOPWZ8-MeaPqGe9SVRU 16499
deepspeed/ops/csrc/utils/flatten_unflatten.cpp sha256=bzx8D2O6b4_rpqlVJAXgLO-dVcZ-ju_L1o7C4sQ1DzM 742
deepspeed/ops/lamb/__init__.py sha256=G9pnfTE9pPPVzCROQFrb0gYsKzJM6ZvoQCpxPQO0Y0Q 34
deepspeed/ops/lamb/fused_lamb.py sha256=s8MnxbjLjejZdurWuB4ohq-zYZmwIalYebgLwHrUMoY 8469
deepspeed/ops/op_builder/__init__.py sha256=KNxn2UYwmcWY2NgTlNXS9ZYwcCpHcnZAqnwprCLtXcw 626
deepspeed/ops/op_builder/builder.py sha256=qpZxx7jFQV0GZQiHoKcCyyOOGKaI5cFQja-cuUnSbwE 9437
deepspeed/ops/op_builder/cpu_adam.py sha256=vcCl8bnOiQxJ7CczkFXNuFMgrIocDD6nu-lbMc_bnFI 2221
deepspeed/ops/op_builder/fused_adam.py sha256=RMoGZ3OtBSpn1wOBPAYEjNQdl00-kVkB1PgU5q2WSoE 670
deepspeed/ops/op_builder/fused_lamb.py sha256=WTtSxn7wLN0EC8bhvMxKZzzTTd_izKcRZy92gyLiWYk 671
deepspeed/ops/op_builder/sparse_attn.py sha256=qqmsYvhcgyr35lfBRciAG6ki-xgs0zVWzg6DRA7-W6A 1200
deepspeed/ops/op_builder/stochastic_transformer.py sha256=OCtIR58omUNSSZU25OUAB4or66e-LV-wZncguoTo0OE 482
deepspeed/ops/op_builder/transformer.py sha256=EOxb-jTWujUk-MeyWwm9SvuE_cQeCKFofjELyaj1VlE 1295
deepspeed/ops/op_builder/utils.py sha256=sZkCyyCMZb5TmShgE65-7u1RSiSlvWk7hkXwRNuMGS4 335
deepspeed/ops/sparse_attention/__init__.py sha256=L9pOegjZ9dbKr2NJ3oUdmHjdW84ObDxsjr5LKNPBM-Q 393
deepspeed/ops/sparse_attention/bert_sparse_self_attention.py sha256=Jx7SDGq_hUmCjXFBb5ZhlQu9jaQNGnFxzd5y7zhYwx8 3570
deepspeed/ops/sparse_attention/matmul.py sha256=L9IYYsCpzVJuNhKzBB06Mzx4Ht8F0KZbCuhU1DGuDT4 29367
deepspeed/ops/sparse_attention/softmax.py sha256=94HME4dpG834n-5m8bG9dpcteoKOVmt2Z9x0pkpWKSQ 11850
deepspeed/ops/sparse_attention/sparse_attention_utils.py sha256=B_qgQTm2raAB_NVJuadtZZrc4BGgAa0pEXGDEIfpX5I 12533
deepspeed/ops/sparse_attention/sparse_self_attention.py sha256=RerRBP5BruPN3xf-bNEOXhLYCU1KOlJUH2826xeGDBM 5677
deepspeed/ops/sparse_attention/sparsity_config.py sha256=Y2GEGwwRs88IBSljnL3vRNAxtEBIP-cCzsNQ7tpTgYc 37461
deepspeed/ops/sparse_attention/trsrc/__init__.py sha256=lQGs6BeeopgA38KRqR18BVUGin1viHz4ufEnjRHmHQo 936
deepspeed/ops/sparse_attention/trsrc/matmul.tr sha256=OUOc9QBVfOIsRlfHUJhLePNPa4yLHIFqP0ugSQFYSA8 6530
deepspeed/ops/sparse_attention/trsrc/softmax_bwd.tr sha256=8HvDg7FWsEepUuZBTI2UZUbaKubjzDcW-OJSQLvlSJs 1825
deepspeed/ops/sparse_attention/trsrc/softmax_fwd.tr sha256=Ay4_cSIYKw_q9yENW0KyuWvwzt7faw_woPSY1hcrrdY 3949
deepspeed/ops/transformer/__init__.py sha256=_kAk9T8TBgCl6dC1SopBVV9X_Wn5wgZPBFybwWFykZM 79
deepspeed/ops/transformer/transformer.py sha256=geRuYWOVtzknS2oQ-22hMNwWZtqzsUIOWmM7Hn7o_bY 22810
deepspeed/pipe/__init__.py sha256=pv3-S3hMqRm6q3v8-aXcisUZa1wLFXgXPozwgmuygk0 68
deepspeed/runtime/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
deepspeed/runtime/config.py sha256=pfiPXtiZEbhY8BhlOZvBps2peeMinK0EgZP87Rb4GZM 27579
deepspeed/runtime/config_utils.py sha256=930TK_n8BNK5HOOQdXwx6tWduSVkRPgpivt9MvV6Zfg 714
deepspeed/runtime/constants.py sha256=GnRgmg9d3YEkVm1m7pVMyJEYJnZe0KbSkhbqimHHz6Y 9538
deepspeed/runtime/csr_tensor.py sha256=mZJsYpLfmiFTXqbGiJpEfEEgs5k3iqbbqhNyhCKX6iY 1999
deepspeed/runtime/custom_collectives.py sha256=W4HFDcHk_7P5nmR0bdsj27E40g77z8EnqI-_HoX1Mf4 5594
deepspeed/runtime/dataloader.py sha256=eF6dzspBaRzOR8ihM7GjXDqfFV0W3VIBXQfYikHDY8Q 3441
deepspeed/runtime/engine.py sha256=8T9BJAdjoKbSxKE1ebkZIOrdMyp7uBxyEPlSEsS238Q 61617
deepspeed/runtime/lr_schedules.py sha256=TB70OHAExU2yhpYMMNlaB0XQd_Wa6vnvaz741iQFv3c 31994
deepspeed/runtime/progressive_layer_drop.py sha256=oPVEoqYrwehbiZGZPR5KJD2CJX7PlW5ynVQ7IHyeNa0 1288
deepspeed/runtime/utils.py sha256=8B3CV89Z4-jqJ_uv5KR3RJ6IA7Ao4s2lZo_O7iMBqd4 19867
deepspeed/runtime/activation_checkpointing/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
deepspeed/runtime/activation_checkpointing/checkpointing.py sha256=tY9zTLa-Xt9Z8Jb92fyapuTlm0AZXGPSFGN2d2eqPDc 27962
deepspeed/runtime/activation_checkpointing/config.py sha256=5ade1g-EdJ9xjwnNh3zZ4SE4cKJJNvL96Rhp3ziV3uA 4125
deepspeed/runtime/fp16/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
deepspeed/runtime/fp16/fused_optimizer.py sha256=5Q-jvohPRO0BlBopdqKRR1hRcVLz2iUXn0dUIFpnV2k 17616
deepspeed/runtime/fp16/loss_scaler.py sha256=-2Oh_DdHz2I5hGK0WzZbTf2_MtV9j6K0mRKqjLsWT58 8994
deepspeed/runtime/fp16/onebit_adam.py sha256=S7eKSnS0MrYSIpAHBN_NBM7q9SbfCmUYTzFSvYiolVY 16363
deepspeed/runtime/fp16/unfused_optimizer.py sha256=FFTLOvEIs10jqck_BXLvSxrz1byccp0Aa8unAZ074IE 15324
deepspeed/runtime/pipe/__init__.py sha256=fbj2OITJbJFReWbmnfjRpPfWTWbOYYA5JZ5A_qB5R6Y 61
deepspeed/runtime/pipe/engine.py sha256=_8UkHoOaBho20MVdrDPHUTnRYSUCaW1iGYcVU8UHtfw 47383
deepspeed/runtime/pipe/module.py sha256=WXG_tQ4-YWPWgcjPGG4h64qeyZG0EGabaLH8JEK3xGQ 23705
deepspeed/runtime/pipe/p2p.py sha256=WS-3kxVg5CQdhmFXd6URcgzqGlF5M991QvHnQMH_AwM 2698
deepspeed/runtime/pipe/schedule.py sha256=5cuqnQb6V6aMfhBPWnC_nwVPtODb9cjp4rUiEO8rHnY 15250
deepspeed/runtime/pipe/topology.py sha256=0G_HY1H3iGzEMcLVhZZyoLHrFmfekZPhTAl68czjpNU 17200
deepspeed/runtime/zero/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
deepspeed/runtime/zero/config.py sha256=EX-C6TyiYQTEIUhj7B3Zl4UdKtIT4T0XcOaVPbYSvWI 3843
deepspeed/runtime/zero/constants.py sha256=LbPmp8Btizq_J0X5pZLmlB9cODPFeL-yOTTHqP_t_iY 3008
deepspeed/runtime/zero/stage1.py sha256=SSggxa6S8Y2GS1I4F5UBvByAT1TjyqQQc9vFO6yDVRc 46488
deepspeed/runtime/zero/stage2.py sha256=tQFXXBJaMEdtbgsUZyztZD8yai8fIi-y9_anWlWzx5A 78763
deepspeed/runtime/zero/utils.py sha256=G6Cbi3ajVdZPCHbdZBRUPqTiSZWDZPfDsZLK3DDxeFc 1452
deepspeed/utils/__init__.py sha256=JKAUs5UqD6R2VFmYWmS083iARxZn1vO59as0LkEMb1c 110
deepspeed/utils/logging.py sha256=C2gJKPAEr18pNi34K5IW2mg-q54EIv01tnVEF17cK54 1619
deepspeed/utils/timer.py sha256=XE7JiCQAsZP6EunBQNuZrsNVLzfrLGJ-W8d1eEoMJSY 6061
deepspeed-0.3.1.dev5.data/scripts/deepspeed sha256=d0mfdcwjFBiukHN3Ac6IJ_zb9kUu2evdoBaPpdT2HZY 92
deepspeed-0.3.1.dev5.data/scripts/deepspeed.pt sha256=d0mfdcwjFBiukHN3Ac6IJ_zb9kUu2evdoBaPpdT2HZY 92
deepspeed-0.3.1.dev5.data/scripts/ds sha256=d0mfdcwjFBiukHN3Ac6IJ_zb9kUu2evdoBaPpdT2HZY 92
deepspeed-0.3.1.dev5.data/scripts/ds_report sha256=Jv11_Wsi-eH7OTh3RIbBUUjKFxYlR8T7kkkH2C6tvTA 87
deepspeed-0.3.1.dev5.data/scripts/ds_ssh sha256=Ryv6GedIoQndSZj3jJ0z3Zq42STOkAdcma4X1rUkJqE 438
op_builder/__init__.py sha256=KNxn2UYwmcWY2NgTlNXS9ZYwcCpHcnZAqnwprCLtXcw 626
op_builder/builder.py sha256=qpZxx7jFQV0GZQiHoKcCyyOOGKaI5cFQja-cuUnSbwE 9437
op_builder/cpu_adam.py sha256=vcCl8bnOiQxJ7CczkFXNuFMgrIocDD6nu-lbMc_bnFI 2221
op_builder/fused_adam.py sha256=RMoGZ3OtBSpn1wOBPAYEjNQdl00-kVkB1PgU5q2WSoE 670
op_builder/fused_lamb.py sha256=WTtSxn7wLN0EC8bhvMxKZzzTTd_izKcRZy92gyLiWYk 671
op_builder/sparse_attn.py sha256=qqmsYvhcgyr35lfBRciAG6ki-xgs0zVWzg6DRA7-W6A 1200
op_builder/stochastic_transformer.py sha256=OCtIR58omUNSSZU25OUAB4or66e-LV-wZncguoTo0OE 482
op_builder/transformer.py sha256=EOxb-jTWujUk-MeyWwm9SvuE_cQeCKFofjELyaj1VlE 1295
op_builder/utils.py sha256=sZkCyyCMZb5TmShgE65-7u1RSiSlvWk7hkXwRNuMGS4 335
deepspeed-0.3.1.dev5.dist-info/LICENSE sha256=mQaUD2Gx8LUz-n2ZuvVReLKAj74RPqUd-_rYVyzNXys 1162
deepspeed-0.3.1.dev5.dist-info/METADATA sha256=CsYIGhMfBaEgrA8ZdzIlHKDpXGAaM6VdW6wlkxzEm_8 1614
deepspeed-0.3.1.dev5.dist-info/WHEEL sha256=YNlcxLoRZfMqjEfLMA4_N5GPCL0Yw1E8c7zrDjEU0Bk 92
deepspeed-0.3.1.dev5.dist-info/top_level.txt sha256=McxMYYNqN6NGtJTUgnUPL645AbNmKV2RPDg_doYKJtU 21
deepspeed-0.3.1.dev5.dist-info/RECORD — —

top_level.txt

deepspeed
op_builder