apex-personal
View on PyPI — Reverse Dependencies (0)
0.1 | apex_personal-0.1-cp39-cp39-manylinux2014_x86_64.whl |
apex_personal-0.1-cp310-cp310-manylinux2014_x86_64.whl |
Wheel Details
Project: | apex-personal |
Version: | 0.1 |
Filename: | apex_personal-0.1-cp39-cp39-manylinux2014_x86_64.whl |
Download: | [link] |
Size: | 5031598 |
MD5: | 17dcaecd4538abe64c4713491c452cea |
SHA256: | bc8c0d252d8053cfabe7736c92fa026a1eb8f2fe55620d1151bf19f50f3886e2 |
Uploaded: | 2023-10-17 13:27:50 +0000 |
dist-info
METADATA · WHEEL · RECORD · top_level.txt
METADATA
WHEEL
Wheel-Version: | 1.0 |
Generator: | bdist_wheel (0.41.2) |
Root-Is-Purelib: | false |
Tag: | cp39-cp39-linux_x86_64 |
RECORD
Path | Digest | Size |
---|---|---|
amp_C.cpython-39-x86_64-linux-gnu.so | sha256=l-ARx-ORiG_2_gXVOjyuyH-sOz1ZbQT-LOhJ8Pv_3p4 | 3266160 |
apex_C.cpython-39-x86_64-linux-gnu.so | sha256=MPzZ9popbyYAGl05C1r1UQH91w2k8QM1jWD-gZtx1qo | 148480 |
distributed_adam_cuda.cpython-39-x86_64-linux-gnu.so | sha256=g0YJPoStx1yD-QtncPx5onXIWShLkNGfApr6RBi7Gvs | 772792 |
fused_adam_cuda.cpython-39-x86_64-linux-gnu.so | sha256=1t1U8Hl-H6QQ6jJBRvrW30rwkp3gEiLikYHrZKy2kkw | 648096 |
fused_dense_cuda.cpython-39-x86_64-linux-gnu.so | sha256=3ZgHFUOCI_PEwzyz1qwljqmEr_0Km6PnB5ouYSLyPfo | 229144 |
fused_layer_norm_cuda.cpython-39-x86_64-linux-gnu.so | sha256=yFgAOEoxK7U6jlKNMvkxOlWlBqnWjFo7Wp-NjRI6HSw | 2067696 |
fused_weight_gradient_mlp_cuda.cpython-39-x86_64-linux-gnu.so | sha256=JSDQ6eDtQjLDv84AeLUgrog01tkHMtY5eAka4UwJseA | 188488 |
generic_scaled_masked_softmax_cuda.cpython-39-x86_64-linux-gnu.so | sha256=DfJjgJ_dyqIEnktIwjGGSAN5HqUAGzn0g4JCHCkzXP8 | 285680 |
mlp_cuda.cpython-39-x86_64-linux-gnu.so | sha256=QOq76-uUXydOFWq_u2Ok2EYp2QH2OdehHPjSR0bV5kQ | 465264 |
scaled_masked_softmax_cuda.cpython-39-x86_64-linux-gnu.so | sha256=r546PWdG9XHA8BwIX8-vz7gKDntv6rcpP5zDLbLhHgA | 1196752 |
scaled_softmax_cuda.cpython-39-x86_64-linux-gnu.so | sha256=h0VNXdoueKa5Jb0T7C8sRMDEAO_gWvMJTcwPP6xtpOo | 2329984 |
scaled_upper_triang_masked_softmax_cuda.cpython-39-x86_64-linux-gnu.so | sha256=qLZvNCx1EDhq8udXUoAvd-V4AudwBp4nvMVsHZtwrzA | 3963392 |
syncbn.cpython-39-x86_64-linux-gnu.so | sha256=VaWx4ojSKirMk1KJY7iOg52uWpjaU0vK2FiLEw7Q-7o | 770960 |
apex/__init__.py | sha256=xqDo4_x2Jj8QLK7bZP2baxu1CmTR5unmIXbPGs5Y5nM | 2495 |
apex/_autocast_utils.py | sha256=aErJc0n0i0aa9FVtrRLURhSvbruk-9g31karwFYYqNE | 664 |
apex/RNN/RNNBackend.py | sha256=JjP4C76g6_AsdrETSLFYhhmwgcTJI04CXvyoL9V02Ow | 11573 |
apex/RNN/__init__.py | sha256=qVTK2fc-idRd-AYL6EyHwurqASi6FQz6zWOz5hSmKS8 | 71 |
apex/RNN/cells.py | sha256=Wxncz7V_yMVglirX6DNR2Mha2EB-IRR1wyC2BWlqcPg | 2548 |
apex/RNN/models.py | sha256=ilFZATY9DNeuncTBiSJjIwFHUuqkDqWGq52VlsS8iCY | 2265 |
apex/amp/__init__.py | sha256=9WR_3lh7wR_iAWbv399MkwfpQFUriKqNSyBYKhiwTrg | 310 |
apex/amp/__version__.py | sha256=KWguYORmSCfoxyQlC64D9LmN21g2KajPIXPH8OTgQ0Q | 62 |
apex/amp/_amp_state.py | sha256=WUzLuon1ew3ii_8fJIItrD2QHJm2b4L9KWVgjfnNgl4 | 1759 |
apex/amp/_initialize.py | sha256=n56mDubuNeBrNqQ-QlQY5oWCKcavbmCKDh9zmev-xTY | 11584 |
apex/amp/_process_optimizer.py | sha256=bAYLePuNUtkpKOC9kLS0BJcgc7RzjJ0WB149ncrcn0s | 20747 |
apex/amp/amp.py | sha256=QXXMOnNk7x8n91tpsr3WWjff5bdwm-MpDGrRyAnDHfQ | 7860 |
apex/amp/compat.py | sha256=V3NK3AGFLtww7iLdsntxeBSVbbJjzCptjZvmWIaa14I | 1393 |
apex/amp/frontend.py | sha256=VhsHNM6LM6bBtODn3FN67JVolMe4HJaP1eLCqHPnXmI | 21463 |
apex/amp/handle.py | sha256=L61fOsnrPJcE6VqP0vYvpkQq56xd5TTSa2iS8LKOpvQ | 12066 |
apex/amp/opt.py | sha256=4-uVH0vXXZyrYAsa7W9bsUa7mVAvLX0ffUx9GWe4cdg | 3446 |
apex/amp/rnn_compat.py | sha256=pm8h6HHWHX7JzQAZQEp2ojAR1sKdO-ZMokvQwGsHFSk | 1995 |
apex/amp/scaler.py | sha256=KSKB5kyqcJP8jCN9CrOML9WRD4marQtb_-dWqPDCN1o | 10494 |
apex/amp/utils.py | sha256=FzxqSNRwZqMfUv0633PIwuJi0IVOZiPnAnHP7HlLass | 7222 |
apex/amp/wrap.py | sha256=KHDHB9Tv1hOn0CPNH6-kPrIGLQhGifzUTufavwRf5uE | 11242 |
apex/amp/lists/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/amp/lists/functional_overrides.py | sha256=qS1-nMYBgbtTrXoNs-X8pK6H94My-yNTKCwrGZIuFK4 | 2248 |
apex/amp/lists/tensor_overrides.py | sha256=u4OJMCkoKLTc_l_-WlljJ4j9LWcecwm0ieTcuk7gFGg | 1402 |
apex/amp/lists/torch_overrides.py | sha256=r8uVEXXHS9YKpyEiD5SsiLFH6q5ZLn2WoxdK54iy29U | 2082 |
apex/contrib/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/bottleneck/__init__.py | sha256=saoP4PGw0pYZEo5d3Cy90ih2DBoIMTQFckvBjd98KVw | 169 |
apex/contrib/bottleneck/bottleneck.py | sha256=84K8_LbWsoS-O7dVyxDdk0GHrOBTVJU2tRClte59ktE | 36558 |
apex/contrib/bottleneck/halo_exchangers.py | sha256=G8MaX2-fGyHwh2REsmx7e4XD3VizIK92e4jCa0MB-xs | 9753 |
apex/contrib/bottleneck/test.py | sha256=9CJJndsbFzXFfeic2NDuH6EPEuvNoK5KOv2Pn470Zss | 3070 |
apex/contrib/clip_grad/__init__.py | sha256=j7N2uJkUffmalgxim4P57Qjn7gnfiKgqeXBw2vGSfnY | 39 |
apex/contrib/clip_grad/clip_grad.py | sha256=9G7S1RiBjArCnASA7mInV2BtGOXh9qdnqYTAluYhoTg | 4360 |
apex/contrib/conv_bias_relu/__init__.py | sha256=Vy6o-vmPJCkr9mxd-cDjPVk0WTXgOJ9Ct6gWgzjN0D0 | 95 |
apex/contrib/conv_bias_relu/conv_bias_relu.py | sha256=IMsgTtQX5njvOJJYCkh2i5PJ03b04rEJF4i7MrZrnlA | 3336 |
apex/contrib/cudnn_gbn/__init__.py | sha256=pRdT9QyPVeAZdpMPWTh-FnMR9AIrSLJDZmvF0n8955s | 40 |
apex/contrib/cudnn_gbn/batch_norm.py | sha256=RB107K-lvpodhRajQlqd8SN1ZBRVR3ASsshh3gin_Ao | 6725 |
apex/contrib/fmha/__init__.py | sha256=_JmsbCGiAb7Og6qd-sS3mllypeqVgdbVoF9iR8byr60 | 26 |
apex/contrib/fmha/fmha.py | sha256=es8XXiKNR7Cafy4SkDBNiw1vbJcT-OW3N3nhylUFE0g | 3577 |
apex/contrib/focal_loss/__init__.py | sha256=gArFqkil8cWAm6L55CNzHy2djfSwnGWB_iyweHYpevs | 272 |
apex/contrib/focal_loss/focal_loss.py | sha256=YotttIW9CF7mzMWPKV3R8NHpHT6IawqD-o5K40qzCSw | 1499 |
apex/contrib/group_norm/__init__.py | sha256=-QDFGb-f1m6s_M1gHB3m_gmJGBImv1Wy8xeOCPwMMFY | 26 |
apex/contrib/group_norm/group_norm.py | sha256=rs9HG9ynM-aPvVbpSGYGFH1ckNtw3tyA1dEa9PqbQQk | 8947 |
apex/contrib/groupbn/__init__.py | sha256=_kvkDd91OnMvsWMcEN1fO227ihXLYlczH0qxuVr8IEQ | 239 |
apex/contrib/groupbn/batch_norm.py | sha256=dbJpNwa0TriWVk17-g3U3BBEB1KtIjLzQUf55ip5gr0 | 11208 |
apex/contrib/index_mul_2d/__init__.py | sha256=vMmDlmyGV_8YR4bxQicp_CDfOZ-yZJPIrCYmfUXjLxA | 39 |
apex/contrib/index_mul_2d/index_mul_2d.py | sha256=DUHfzdrBCcKhcEkmv77SNUHS32b71ZBMpiwirLWtc48 | 4594 |
apex/contrib/layer_norm/__init__.py | sha256=DqcoXOA8-vhzossUcqj9VvINwuPMLTEvHaTBCnjRMyE | 38 |
apex/contrib/layer_norm/layer_norm.py | sha256=LtEC7MWQGgwkCTyXLgQN8VpCnP44qGM9lexbiDPeW48 | 2078 |
apex/contrib/multihead_attn/__init__.py | sha256=H9z7PpWRCoXDhJUn4gAlfaO9PeWe2O4GGaFcObr7A0I | 176 |
apex/contrib/multihead_attn/encdec_multihead_attn.py | sha256=puVf4S6_-fkyeIiX5vHZn7p3W9o1ppQkImg_n3yTpdw | 7538 |
apex/contrib/multihead_attn/encdec_multihead_attn_func.py | sha256=4hva6MsgGHE8Bm2tQYnuGRMZ4JfI4bzLecuD3U3E7F0 | 16844 |
apex/contrib/multihead_attn/fast_encdec_multihead_attn_func.py | sha256=Z58znUuJpI-8l3soaxH8bMIsbujfeonX53JxSdY4wM4 | 2974 |
apex/contrib/multihead_attn/fast_encdec_multihead_attn_norm_add_func.py | sha256=A9Yrj7Q_nyjRiO2npIPSZ1UInwNsmh66p95P6yqP7g4 | 4258 |
apex/contrib/multihead_attn/fast_self_multihead_attn_func.py | sha256=MS8D29Mwx7RoLA7BLcNUM22Gvl8AlKJRCajFtByIvYY | 7679 |
apex/contrib/multihead_attn/fast_self_multihead_attn_norm_add_func.py | sha256=laMK0qVGRGUMhfOjv-VrrGqUy8Cie7L2vhk_U4Zkrgo | 3492 |
apex/contrib/multihead_attn/mask_softmax_dropout_func.py | sha256=Hmymtq7TJP2Zu6CUUYbVqiBUY5e0fqEteTpn-BHvHqw | 2456 |
apex/contrib/multihead_attn/self_multihead_attn.py | sha256=heLGoAKY4EOAXcr0S0qe82h22b6JghXNubZe4qspBS4 | 10085 |
apex/contrib/multihead_attn/self_multihead_attn_func.py | sha256=zAfvxWtWKsFJAWk_EWckZ7nztX0YJ39TnKQgt5sC8Jc | 14144 |
apex/contrib/openfold_triton/__init__.py | sha256=kaZ0wDbNk0CIWEuvS_58m_WEYClB69IZ2oya12W5NlQ | 3781 |
apex/contrib/openfold_triton/_layer_norm_backward_kernels.py | sha256=q2IwZycRDPDYvG4-NGPEr35IKzIWOTGbWTEP8CP_lz4 | 10560 |
apex/contrib/openfold_triton/_layer_norm_config_ampere.py | sha256=hZnNvLNEgPnCKLYb6RmNZzH4C0AQzIQsspSRpUpFSDA | 5981 |
apex/contrib/openfold_triton/_layer_norm_config_hopper.py | sha256=yPO0DYkueV_KtPmiW2vdu_2hWoCU6YblCHpHjPVqTpI | 5984 |
apex/contrib/openfold_triton/_layer_norm_forward_kernels.py | sha256=Nj6Oy_8d1Sx8w9OPGPMo3OU1mtIVP_sMZpbSqibEExE | 3939 |
apex/contrib/openfold_triton/_mha_kernel.py | sha256=JCfvGXE30gJkf2AVZebqmpe6wVe2Dudr0b_6djxdoeI | 21537 |
apex/contrib/openfold_triton/fused_adam_swa.py | sha256=ejWtBaGw8Dc0w8aEhpcT-c1K8kMdy4Tzq4-cyikoTNo | 18957 |
apex/contrib/openfold_triton/layer_norm.py | sha256=D9BWjdhorJt3RehzlX7w9C8z6vTHClNM60_PCANITmw | 6954 |
apex/contrib/openfold_triton/mha.py | sha256=Akvvaa-I00Pw1YgmTA8CaTEToubcQL2cvs4OtW2YddE | 14244 |
apex/contrib/optimizers/__init__.py | sha256=W1O4yG5lTVenYQk0_SrPIXIFvG72ZBto8XAiTbaQFtg | 111 |
apex/contrib/optimizers/distributed_fused_adam.py | sha256=An11znVLdwAZ3vDvoAqWqTkf54kG0M_PGTg1c3gnIiY | 123577 |
apex/contrib/optimizers/distributed_fused_lamb.py | sha256=nCzbkOwaO3YECRRnuFGO2IdKDwn6y6vnxkb2-MBHzNE | 56980 |
apex/contrib/optimizers/fp16_optimizer.py | sha256=irUrHVYJiQwPZp2nTCz42TIyAnWXZ-nw1_Bn7iWS2MU | 10448 |
apex/contrib/optimizers/fused_adam.py | sha256=EcPhaTEQ7x3-kgXVvtVfc7_POOhMwUJlpBQtnBx1yFA | 9284 |
apex/contrib/optimizers/fused_lamb.py | sha256=O6hUkV7ADWaD11VuYK6a5X54K044kswrTMQzTsQ2ipI | 9408 |
apex/contrib/optimizers/fused_sgd.py | sha256=Ixlrr2t3RxE2dYLcju_x-KaFf22A-G2ZiUklovKaqi4 | 9468 |
apex/contrib/peer_memory/__init__.py | sha256=7qtSoyIX-C3sSnNaEITgF46o9CPgsZ2FYqW1D7GHqCE | 97 |
apex/contrib/peer_memory/peer_halo_exchanger_1d.py | sha256=6M0OcgdpvciPHadhhl5-4Ezua9QOoeaZvJy0F_5tbic | 3849 |
apex/contrib/peer_memory/peer_memory.py | sha256=IOtxfGxxCgKTWmpKpruh5PbgTTpXKNAoha5NCWXaQyA | 4748 |
apex/contrib/sparsity/__init__.py | sha256=tvwhMT6ERP6QJh-527FQ2Gzjui7IUD9FNevukWtJSdE | 61 |
apex/contrib/sparsity/asp.py | sha256=TiCYXUX6kbSU7b388nX8fJV7fc5jbprMbHl8k4LYDlQ | 18456 |
apex/contrib/sparsity/permutation_lib.py | sha256=vjJYWagPMqAzjH3GzqI2G9lVgMXHPnkSx-sSBV4Tpr4 | 89189 |
apex/contrib/sparsity/sparse_masklib.py | sha256=h5G9aWSO_8WQL8JyUJ8DaA-0Y9_lWDgClVuH8Zt8adA | 7422 |
apex/contrib/sparsity/permutation_search_kernels/__init__.py | sha256=a3MQBqMtnwwfwQLKJudeHAkn6FgV3kzjiEJXIjAcugA | 136 |
apex/contrib/sparsity/permutation_search_kernels/call_permutation_search_kernels.py | sha256=be419te-WhNQZ7CQfD6x3BJwneHd8nd64Hkemnywrck | 3912 |
apex/contrib/sparsity/permutation_search_kernels/channel_swap.py | sha256=iL7lKtIPU8ZFerLwchMqZsBxzHwpuNE3DmNseB0FflI | 8516 |
apex/contrib/sparsity/permutation_search_kernels/exhaustive_search.py | sha256=PcFfF_MdWKZIyiC9vgR_gH2CtE00G-D90OD6KZ6gBlk | 17106 |
apex/contrib/sparsity/permutation_search_kernels/permutation_utilities.py | sha256=8Iq9ScXrtg4rObf3h1SlR2F9IstZ8LMjm9ifcPk-S0M | 21441 |
apex/contrib/test/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/bottleneck/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/bottleneck/test_bottleneck_module.py | sha256=vwOyjHTmlVcn0gpHEjyOrTSl731jNni1n6f2Mcynl04 | 11569 |
apex/contrib/test/clip_grad/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/clip_grad/test_clip_grad.py | sha256=vRqS9oDVDeWC5YvOvmrMmdRe89CcPK0zbbGLyLHyGuQ | 4983 |
apex/contrib/test/conv_bias_relu/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/conv_bias_relu/test_conv_bias_relu.py | sha256=QUvppscCR4Y-Vo9kpE_CglHe3euSMUR84v2W9kE3U4k | 6572 |
apex/contrib/test/cudnn_gbn/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/cudnn_gbn/test_cudnn_gbn_with_two_gpus.py | sha256=WVaxadoO9iJTGKgknOM7t5JNIJxWLMNWsW3vaYqOt0Y | 4902 |
apex/contrib/test/fmha/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/fmha/test_fmha.py | sha256=Mm7lcpq_X_u3Q8z8XBHq3mXE3K42iTBUTrQ0Fgd3OQU | 5412 |
apex/contrib/test/focal_loss/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/focal_loss/test_focal_loss.py | sha256=74aWJZ6Fac6J0YzjBv9_IvWwXnaJ2-znqqna1FN5ILU | 2253 |
apex/contrib/test/group_norm/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/group_norm/test_group_norm.py | sha256=iWeI3TdMSyjAptYP7CTz0QU3Ba5na_pmcFf02gTgpg0 | 6641 |
apex/contrib/test/index_mul_2d/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/index_mul_2d/test_index_mul_2d.py | sha256=Hf7KXKpFNpXRzfTXWX8CSEvRhFM1jnS5lPnosXorHt0 | 4337 |
apex/contrib/test/layer_norm/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/layer_norm/test_fast_layer_norm.py | sha256=PjlCZnOlAU0EDs89CtE7JH5HyXeSqYKBEBxUalWrwyI | 8464 |
apex/contrib/test/multihead_attn/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/multihead_attn/test_encdec_multihead_attn.py | sha256=InTEJGSBymx_bmu9tZw3ohw2JQ7KBrb5V40Y14GPd2I | 5890 |
apex/contrib/test/multihead_attn/test_encdec_multihead_attn_norm_add.py | sha256=yRIV3a7ekgZLtzMqWaJBi_IccTj2vd9BUHLNqcxUMbw | 3154 |
apex/contrib/test/multihead_attn/test_fast_self_multihead_attn_bias.py | sha256=QaBb063CYJNWLb53dZqsBzDMlOOgIuJEs5cI6EnN_0c | 2898 |
apex/contrib/test/multihead_attn/test_mha_fused_softmax.py | sha256=Qh2cLh9TJKnIzJtOusi6d0jdzC4Qt1Qjko36V35msgs | 1899 |
apex/contrib/test/multihead_attn/test_self_multihead_attn.py | sha256=rC_WKqOti_UwG5ZnaFdQK_E4XFf9cUysz8H3I_oFJ0g | 5188 |
apex/contrib/test/multihead_attn/test_self_multihead_attn_norm_add.py | sha256=D8XaS2s5R7smhZjLZV4miv0QB8BP-cXT3IYfRVxLlJs | 2615 |
apex/contrib/test/optimizers/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/optimizers/test_dist_adam.py | sha256=N6m1N2wAWTZH_3zKyZpPs-IpI3CcTHc2pgv_46egu5Y | 27513 |
apex/contrib/test/optimizers/test_distributed_fused_lamb.py | sha256=mWYZ73BqmRhn9ap8KyrzegCuiXLAhKHG2jXYXgeZYAw | 5013 |
apex/contrib/test/peer_memory/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/peer_memory/test_peer_halo_exchange_module.py | sha256=TaxNGgqezgb_9nNtQQg_TRjU_4xkqK88v6igrlfO794 | 10131 |
apex/contrib/test/transducer/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/transducer/test_transducer_joint.py | sha256=NZk6nyZsQ-Mvt0kV8QHt_rQECzXVbkDsfbvLL9PVHiI | 7399 |
apex/contrib/test/transducer/test_transducer_loss.py | sha256=F3jIaKlnxi7kpKaMz46Lhv9sblq7WDxnQswxHsEnmGA | 7111 |
apex/contrib/test/xentropy/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/contrib/test/xentropy/test_label_smoothing.py | sha256=rbu_1wdtl5jDGo3nXuwPfNAGoWcff_uHE3oK9uFZttI | 5037 |
apex/contrib/transducer/__init__.py | sha256=5pySN_45v6NtEvPabUpl5CDG_NFR_4ajyXSotIEfHaU | 112 |
apex/contrib/transducer/_transducer_ref.py | sha256=X3nJUg-ZTNk8wPCjnKhJ-Uc4V2bwKrwy28T702id9Q0 | 4730 |
apex/contrib/transducer/transducer.py | sha256=dp_u3fTTLdn8iIvHOMGwac3U222tL_T4GJFsXADaIvo | 10129 |
apex/contrib/xentropy/__init__.py | sha256=q_D3Bfdb6LqrG_tbVsXmiiTdKB6h6-QO51wxjtrOoRk | 101 |
apex/contrib/xentropy/softmax_xentropy.py | sha256=nyNbNe0Ij_Cq6c7kcWvh5VEFnxtWLdn7P8J4XZyPhtY | 1025 |
apex/fp16_utils/__init__.py | sha256=a4bawbJatpFpogmV3xfMzBrTzcnUOl8PZKn9YZhPuHg | 367 |
apex/fp16_utils/fp16_optimizer.py | sha256=EmG_SQ7uMMEJb2S9P8wZwWwfQ-ovogHkHxxOYDsCUt8 | 28438 |
apex/fp16_utils/fp16util.py | sha256=FXDPOxE4ANQMmZWtGs8Br4PulBLSk9ljBZYMrCNKH9g | 7351 |
apex/fp16_utils/loss_scaler.py | sha256=7rFZctFWxwQoHqO1qBa7JXl5QNcnNtZDLED0hIOQ1Jk | 7740 |
apex/fused_dense/__init__.py | sha256=Ylk0cw0NVvOYP6okBLzbZnOhYgJairhLpNU-0JPK2pY | 27 |
apex/fused_dense/fused_dense.py | sha256=EtViMVLfJWBNLbuq9ub2EYngLq0SgpZj9S0222zooGM | 4072 |
apex/mlp/__init__.py | sha256=1YabSAf0kHAGB5NUowMvh2jyoOli498rYowXZPpzOeM | 19 |
apex/mlp/mlp.py | sha256=wVivxgtc_FrVjwTNw83erpFwknGsK8aOJBFn7s3JpZc | 2757 |
apex/multi_tensor_apply/__init__.py | sha256=h0AXmX6jg_bPAaCZIs9StdKxQl7YMVw4twAGdRxGtEE | 100 |
apex/multi_tensor_apply/multi_tensor_apply.py | sha256=ISd05Wp8rIEXLqHILt6nfE6J6tKQpjjqt6Lq3tL4Hjw | 991 |
apex/normalization/__init__.py | sha256=CEl1JTpMC63hZNOebTF5BqHWJqluDkgePJvn3u2Gfl8 | 99 |
apex/normalization/fused_layer_norm.py | sha256=xba6pwtm1C150yOtEkOLpSu7v28Uja5H3r0fGay8XYs | 20509 |
apex/optimizers/__init__.py | sha256=bl8PSr7vYEwJiXoQ8eanjjp39al3iBI3ilJBQdC9_5k | 246 |
apex/optimizers/fused_adagrad.py | sha256=oVsdFbCltYmWi1T3gKq9Zd0WY9m2NdMfhPzNFgqJ2zY | 5231 |
apex/optimizers/fused_adam.py | sha256=LhZi3FK-i86bB1S0ebG8vHDvU8X6LSiUCVdhKIbbmOk | 13941 |
apex/optimizers/fused_lamb.py | sha256=aOcaCypZYumUPTqhkVVdFgZzgRWtZk-kBx7Z38Cw_7E | 9910 |
apex/optimizers/fused_mixed_precision_lamb.py | sha256=-A4uKRUpZmcKkkZVtYMuSteaFw-wFpGUn9n9hbsFCC4 | 11275 |
apex/optimizers/fused_novograd.py | sha256=Woq81SUwTdjbApch2ExTZkbl83rbiWVIwgla_mU_HQ8 | 10652 |
apex/optimizers/fused_sgd.py | sha256=hRbSbHOomWOVk_6evy3z1feXyB_nm0xl15uOICt_rYk | 10041 |
apex/parallel/LARC.py | sha256=PVyYNuSnlPipv-IW55Z1NrCX8B0McL62Ph8bNsFJqJ4 | 4018 |
apex/parallel/__init__.py | sha256=UttQmatLmv3DOvygGiFds61cypWJK_yeKLV0R-KszRw | 3884 |
apex/parallel/distributed.py | sha256=BwEbnge0Gqn0nGl8McGZtAj2_pzaEYWQXGs8m1M0UY4 | 30828 |
apex/parallel/multiproc.py | sha256=y47J1SdFZiJmVF0T-XPZZjXZqov0rHzaG6PG5aGqTc4 | 884 |
apex/parallel/optimized_sync_batchnorm.py | sha256=Sut3oqsGz_ZIppDyaACfMlN1VlgiyWJDflLZVf71qa8 | 4364 |
apex/parallel/optimized_sync_batchnorm_kernel.py | sha256=4l3M7I4XL_FgqOw13Yw_NZVqXnCt4adJ7rprDQ7oq5c | 5467 |
apex/parallel/sync_batchnorm.py | sha256=AGfX0xzPJIgeB89QZE_0RHZ7mgNu5AC0f881LywT_UM | 6727 |
apex/parallel/sync_batchnorm_kernel.py | sha256=eA8rI9w8IhZAZZ9W1sfv-7N0yr8IofaQphGocWZ4054 | 3761 |
apex/transformer/__init__.py | sha256=Ukt0WT86fst1JupT0JxW2dq17KHMAtpctzsVG7eMMKA | 577 |
apex/transformer/_ucc_util.py | sha256=1gHgWaPFia63iX9-HZNBbyd5Y9Ed1zFkf7dzPJaRfDA | 232 |
apex/transformer/enums.py | sha256=KxUW5IvHr2JkckeR5wnA6EQAYpHeMZAHDBBIAydpkhY | 914 |
apex/transformer/log_util.py | sha256=zrg7YUTuwiDc2AjOIiKaXh0-H5JophjvmgT6-RUPFAw | 371 |
apex/transformer/microbatches.py | sha256=eY3fB5iua6qFJKOXgt43pUqixZdweNTE5fg1IGnjcWY | 7173 |
apex/transformer/parallel_state.py | sha256=KIwMHDPP3PhqPPJX98SKvojNGdhT8QfZMxBG4HOX9XQ | 33070 |
apex/transformer/utils.py | sha256=4p6sQ37wHLQy8YUj-KuoC9MeHzfwvcS_IFDczYIDbD0 | 1923 |
apex/transformer/_data/__init__.py | sha256=Ac3g8HeSOCB2mfBZR-g-HSgdSVW5RQ5Bc2F4tTIV1co | 248 |
apex/transformer/_data/_batchsampler.py | sha256=QjW20ypl6TP3i6eXrav9tlKan27116qPNCqQSsxh0Cw | 7203 |
apex/transformer/amp/__init__.py | sha256=-2Iz57655leb41sWVNPwMJ0si98GdwAJCxvuTBJkR2g | 90 |
apex/transformer/amp/grad_scaler.py | sha256=LiYm-Hyg6kcXFKNnLLTmMLZijFeKxSeYeOewdhS-HNc | 5577 |
apex/transformer/functional/__init__.py | sha256=DXHMX-_3wTK4l6Y2ZqHF4gjcgnN-KqwOffLEWgAkG8Y | 120 |
apex/transformer/functional/fused_softmax.py | sha256=5UIqreelmxpuly9XXz_UQ5nvsQtRA1sixX_kh2wgosk | 10915 |
apex/transformer/layers/__init__.py | sha256=tjdITvIyvmi55tss8hz_VKcLdBKOHDO2WXKh1YMLxdY | 339 |
apex/transformer/layers/layer_norm.py | sha256=STqG4yM_NjU_L616s5shRH6cWGmoeylM8-ImQ7v-6ac | 3456 |
apex/transformer/pipeline_parallel/__init__.py | sha256=DgNboGU8-PL4-0eayclmBoJySI_zDjUmnaWb6NKCB2Y | 227 |
apex/transformer/pipeline_parallel/_timers.py | sha256=2mqOYmSlnfYCOP9Fn3pb-uXkRiUFjJ_p0JHuFumnI4U | 2538 |
apex/transformer/pipeline_parallel/p2p_communication.py | sha256=u6OvKktZyyEeGY0dbiqZKlpY_uEJ6_oncmKhuGYFPGQ | 28348 |
apex/transformer/pipeline_parallel/utils.py | sha256=rLO6O1uZCgedTHpHMghNK2zccJgLNOFs95JNhCk4Djo | 12610 |
apex/transformer/pipeline_parallel/schedules/__init__.py | sha256=BK3As2aOXGeEZVRYxPmGaBPTrDf6H-8h6E5Hr89x7SE | 1399 |
apex/transformer/pipeline_parallel/schedules/common.py | sha256=tcEq357szICa_IjCkFW13HTFp2VBOhSLz8VzGgoONIE | 15829 |
apex/transformer/pipeline_parallel/schedules/fwd_bwd_no_pipelining.py | sha256=9zYLGfNUI2zSaR7vXXJwKZJox5MnPFW4YfJv6PXQeAs | 4663 |
apex/transformer/pipeline_parallel/schedules/fwd_bwd_pipelining_with_interleaving.py | sha256=GIcDt6LWX9n2g1PqdEomI1QSqo_nUZ2l_WsNh6nAyKs | 35282 |
apex/transformer/pipeline_parallel/schedules/fwd_bwd_pipelining_without_interleaving.py | sha256=G8UIcENtxQykJ-U9O--HcAkCTt2t2oD2fF3lWwoS4AM | 24375 |
apex/transformer/tensor_parallel/__init__.py | sha256=zQfh-Xwqc6V_7Nc7LB4HWG-pS-HX2bbMrBRIzHbdR-s | 2514 |
apex/transformer/tensor_parallel/cross_entropy.py | sha256=WzaljIF9vGc7hg_Svce5fxbISwwW2UBx1mmYydoiIQ0 | 6446 |
apex/transformer/tensor_parallel/data.py | sha256=nspJ6tyUoCJA_oWOcPHrdp3Eg7NDlhNkyDmjgaGjFI0 | 4047 |
apex/transformer/tensor_parallel/layers.py | sha256=g6guj4pC6DNl1noQsQ147l89QA8AX0jXngT_XukaftU | 31021 |
apex/transformer/tensor_parallel/mappings.py | sha256=oNdbThNZG-mU-u0VP7Qrp-bDF2NJuCf7FDRNERW-AFc | 10918 |
apex/transformer/tensor_parallel/memory.py | sha256=0drPj3DDbs0dNLigyysEeLATJMLi5eHaSzzOf64IcDo | 5354 |
apex/transformer/tensor_parallel/random.py | sha256=9eP7EQ75BJ2qs0gd5vh6pPNB0HbbPdF82Mr_rGyLofQ | 11852 |
apex/transformer/tensor_parallel/utils.py | sha256=55dg7IqfKEW7X3lhR2ndWXtF_JdfOwn3X_3s2zORHUU | 2396 |
apex/transformer/testing/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
apex/transformer/testing/arguments.py | sha256=LqGB5RSgzuVDpirGnXVKMFeLqMrYwgBjE3ca6DQSWug | 50649 |
apex/transformer/testing/commons.py | sha256=WAPzJ8vkgB4Jul6KmjrIOG8XZkfaQEew88Y9h5srRqE | 9682 |
apex/transformer/testing/distributed_test_base.py | sha256=X7E9XZ2zhliSilwNxmsZHLeNL9NX5JTNi_CiYxu6aTo | 4008 |
apex/transformer/testing/global_vars.py | sha256=6kLrOlaK5Wt37gDCx-BNBbBu0-SA7IIZ_SYk6e7BbjA | 9009 |
apex/transformer/testing/standalone_bert.py | sha256=f8gfTA6vbF4Q04oSJPF8SW8JQviVTH5fQtxcnQ97oBw | 9945 |
apex/transformer/testing/standalone_gpt.py | sha256=-xHN2b2_303FHqBQdxXSfdM3uYBcU9nH102goJ5rEOU | 3935 |
apex/transformer/testing/standalone_transformer_lm.py | sha256=bC1mzCVpiZcyWcn_C_AhXwzwYXk9o_kVoi3NMhfLNxE | 60815 |
apex_personal-0.1.dist-info/LICENSE | sha256=iCfP3zpgJ8FCvnf3ysuQwaSVG9LZSMgZeEznqZOGkK8 | 1449 |
apex_personal-0.1.dist-info/METADATA | sha256=68CBGpzTKok0S7R4AGoLILUSEGlXzrdgdBzHuM9vPWU | 211 |
apex_personal-0.1.dist-info/WHEEL | sha256=_95Z4mhUU5FheowRQnEDZA-5_73DZ-fJ6DzrSduwonc | 103 |
apex_personal-0.1.dist-info/top_level.txt | sha256=E37dtltv5_1Uo4E2M2OADQB7YJHFr4Hb1twcTBudX1c | 264 |
apex_personal-0.1.dist-info/RECORD | — | — |
top_level.txt
amp_C
apex
apex_C
distributed_adam_cuda
fused_adam_cuda
fused_dense_cuda
fused_layer_norm_cuda
fused_weight_gradient_mlp_cuda
generic_scaled_masked_softmax_cuda
mlp_cuda
scaled_masked_softmax_cuda
scaled_softmax_cuda
scaled_upper_triang_masked_softmax_cuda
syncbn