apex-personal

View on PyPIReverse 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

Metadata-Version: 2.1
Name: apex-personal
Version: 0.1
Summary: PyTorch Extensions written by NVIDIA
Requires-Dist: packaging (>20.6)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 16 characters]

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