fjformer/__init__.py,sha256=UvTcoMLrx7fSixXqfTE1qP9IaGDnZbFEX-4H9wfGa0w,1365
fjformer/utils.py,sha256=3nLxe5H2Kr7wpyVh3DL2Rp-kk5cVchUHw0Ox_KZGCPs,5224
fjformer/bit_quantization/__init__.py,sha256=zJy93z6wjc2ssuMLrt1dEUsCgW4lOTwNZn4sAmI0M9M,1052
fjformer/bit_quantization/calibration.py,sha256=B1afi1oY0D5FMIyx1PKzmo5Tm0A6L2jfxPcQCU3xQ7Q,1748
fjformer/bit_quantization/config.py,sha256=hUiEoKsb6GFYZVenU6ZLRe3YUGmon_6e4AJt61zAe0k,17573
fjformer/bit_quantization/int_numerics.py,sha256=HYvHUxgBT49Gke_MDmPjegdkddoXEQ-0cH-eGkefMps,3803
fjformer/bit_quantization/no_numerics.py,sha256=jCarC6G8bxRZEkERltYXDCFZAPm-LDpwSBMDBSO1ZN8,1354
fjformer/bit_quantization/numerics.py,sha256=JzbDlBR3xVRsdF3kWMqMra0N-417w1VGivgfmjs9yPc,1457
fjformer/bit_quantization/q_dot_general.py,sha256=EGUeTyoFv3HVRDJ2k0BNB_EOhW2T6SLfm8-wOXKEajs,21480
fjformer/bit_quantization/q_flax.py,sha256=tpPXRubla6uqF2U1j9--vQUSm18FrCclmNSnvKeu52E,14725
fjformer/bit_quantization/stochastic_rounding.py,sha256=qDiulQTPgg4eBrPWvWdN64qJQkjlzvsjzOueK-LBXgA,1495
fjformer/checkpoint/__init__.py,sha256=tIyFsRkcJHNxIe-k8qR0UW6OJhYa14G5GmAfjhgmKPg,377
fjformer/checkpoint/_load.py,sha256=sd4Hqlw_LDGiVZ23gUNSwaRDBNAENlu0ZBNQYn_sf9Y,3254
fjformer/checkpoint/streamer.py,sha256=y501kjmlqYwDCZlZKPP5eFGUx91K2tAXL94e-pRi5PA,13726
fjformer/functions/__init__.py,sha256=T4TP0ghMK1tCYsGX1dAXq_e0MPRfKBa6WyN_E_Rc5Vc,1828
fjformer/functions/_func.py,sha256=7fnJnrdYiy0wOPyAWiy04gCfnGIKbefGbbBqqUWx_9E,2050
fjformer/functions/loss_func.py,sha256=-MfT9jNxgF_kY9uK-KU0MjALWraxbgW7ti_5LPSE8Bc,25746
fjformer/linen/__init__.py,sha256=ZZleVUpJ03d2KV8NQZUF9C2Ouy45YK8Jxz00TRei1NU,1238
fjformer/linen/linen.py,sha256=pJLEvMlM2FGMuLp5bg5UR0Te4Ht_qp1aNaF-co9YDYk,72021
fjformer/monitor/__init__.py,sha256=k3j-Gk2ZFAnOAOqi-R1qbabFopPMnZPdP-GLkvGf3gc,392
fjformer/monitor/tracker.py,sha256=D7nXWhCYiYgB5f93SDtI1WTBS3u0uyCD8Ep2R33AseI,6920
fjformer/optimizers/__init__.py,sha256=-S9VJoWKbyCBi-eW3sRAer2kTKotPuZIbLZksPg1lhs,2459
fjformer/optimizers/adafactor.py,sha256=H0E3z0Hd3P6a08fY9YCbs4k8vinS2PVcw0Jz_PfNmw4,8906
fjformer/optimizers/adamw.py,sha256=ikLsEgI7RArqs65XjTxVc5j-Sa9GjB1Pne0q99vBS6Q,5983
fjformer/optimizers/lion.py,sha256=MPKcxFpPNqAvqeaZAscS6P63rI2q_s-uJpsHnKbsTqY,6036
fjformer/optimizers/optimizer_utils.py,sha256=trI8jIUeUc-z1SxX0DIOL1Bw2yDYezQBMpZquIFr8eM,1064
fjformer/optimizers/rmsprop.py,sha256=Joq6ofvFN1xw3bOniXFZpVEKIi-8YQfkq1dm0lmIr7U,10362
fjformer/pallas_operations/__init__.py,sha256=N_5-n4fTp1syuKXSDnEu49A9BqlQD6psAMLInZ9YXvY,169
fjformer/pallas_operations/efficient_attention/__init__.py,sha256=wzo0L7EtaQXLxlYY0BH8Zqv7WU3vkr4fsZeCNS8hCAM,135
fjformer/pallas_operations/efficient_attention/efficient_attention.py,sha256=zV9Phfv-frTXveYUq7Uirepr-DFivmj0JoiN5bgptPA,8042
fjformer/pallas_operations/gpu/__init__.py,sha256=5hB9CeoI7Abbq-qoMR8pyld1bISkcQ8RTlcCiDkFbuI,196
fjformer/pallas_operations/gpu/flash_attention/__init__.py,sha256=IUFOX-3_w_w0X_xipZPCaX0F29UbzTwtmAs5wqDdbGE,118
fjformer/pallas_operations/gpu/flash_attention/mha.py,sha256=pqSC3NtTOSCWSTJ3A1_hW2DlACPbKmr0QfYMCP4J76s,20056
fjformer/pallas_operations/gpu/layer_norm/__init__.py,sha256=GZUObViM0z12DoY6TiaU7ExCkVoLh8B6d3lDT87A5Yc,103
fjformer/pallas_operations/gpu/layer_norm/layer_norm.py,sha256=bdAV6k6Y1_CKUo7tSZgsoxineysRKSLERyjfBIfpL1M,11172
fjformer/pallas_operations/gpu/rms_norm/__init__.py,sha256=60EOSi8rzvLljpZX14UInlAKUtRt75L7dn_UeH1yY4s,95
fjformer/pallas_operations/gpu/rms_norm/rms_norm.py,sha256=AOILYwxQteigOFsTwXwmsIW0tS1qL7wtmX6pksOpazo,10012
fjformer/pallas_operations/gpu/softmax/__init__.py,sha256=6bbYRHqm55L_sOPkoXGbXCK1ybyJjyc1uyL55n4QqZ8,91
fjformer/pallas_operations/gpu/softmax/softmax.py,sha256=ZrMlySJ8HQoYm--CZIW6PLgt9IQnuOjB-zoIMWoMJis,2285
fjformer/pallas_operations/pallas_attention/__init__.py,sha256=aD9Jp4ymMVBjackCIpliWhYzmuiaQz3ceYC3BMcsDs8,114
fjformer/pallas_operations/pallas_attention/attention.py,sha256=zycki-6m8vqwzN0M-ChfgnItffv455seJal0m3a4KiQ,16170
fjformer/pallas_operations/tpu/__init__.py,sha256=DgHKWrV1CelNqB3SFEIfAMSTZa0288zJxlEECao4_yw,236
fjformer/pallas_operations/tpu/flash_attention/__init__.py,sha256=HNMA3JHsoCKhC_kNXcbyGvCCUe5PkbDMZknxSUVAcOY,148
fjformer/pallas_operations/tpu/flash_attention/flash_attention.py,sha256=2ta9lL_oaUGGm6EAGe4CPXg4CEwvLuY8i5V4tBpolTQ,54910
fjformer/pallas_operations/tpu/paged_attention/__init__.py,sha256=HBmn5dh3hwJFOfgdynuDEEAfLc-_QHdU1n7Ps-ZP500,184
fjformer/pallas_operations/tpu/paged_attention/paged_attention.py,sha256=xBkT76r7r2yRv4MWo0ky9ai-B5vt-KK3hnQSFNayJj8,24589
fjformer/pallas_operations/tpu/ring_attention/__init__.py,sha256=lh_n5UA1OsshQ4WkAiAI2wuS_A6IdGJPbCAlEZ9Q61c,315
fjformer/pallas_operations/tpu/ring_attention/ring_attention.py,sha256=OwHIKxj5MHj3Y7pGvht4plSC8dnpIR4eFmSFdnsbXV8,75385
fjformer/pallas_operations/tpu/splash_attention/__init__.py,sha256=X8v25LDF1lyub3Qwt2oLKLW8oB4QGxmIXm08n-ZygvA,2008
fjformer/pallas_operations/tpu/splash_attention/splash_attention_kernel.py,sha256=1hyVDVB9ffqsQhIsTe5daOA55sujJBjwGRhucqGPpRI,80719
fjformer/pallas_operations/tpu/splash_attention/splash_attention_mask.py,sha256=9DsaxSEYwK2rD5F7zk8Nbwn-ozz3j7Ip-yJsTVn2B8k,16217
fjformer/pallas_operations/tpu/splash_attention/splash_attention_mask_info.py,sha256=OZphQnxr-dZWcT2z7Z-y86XRsTa_e_I7Ex9b3TkpydQ,37092
fjformer/sharding/__init__.py,sha256=1AaiEyMcjs3Qpbi18SUEB2mH2QGY7P89ygUd2OoFz7s,378
fjformer/sharding/sharding.py,sha256=oYUTTbJmaNkD6t81TPe_kv-6E5cToUDYJh-a1114HGQ,10915
fjformer/sharding/t5x_partitioning.py,sha256=ePjN9WQLK1iH3-O6T56d2eSvTYRwUwqYsgl68DcGB6U,46436
fjformer/xrapture/__init__.py,sha256=5NPeL50SZGZg1P0h_weNp0OCjun65anBKsQPUpRJlNQ,1048
fjformer/xrapture/implicit_array.py,sha256=SXXIlwye_ZHrdSnAuxCDjtXTzFmWR4zZ4hs_Q8KULO4,37155
fjformer/xrapture/tracer.py,sha256=ctCWyEII5yRPgjd86cEOiVQdGJmUZfbQsAyfnNbKp1I,2937
fjformer/xrapture/xrapture.py,sha256=1kBap1fdROttPZURMQyGd1OmZ8vXeYuAFOiwfYfmqoM,23612
fjformer-0.0.67.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
fjformer-0.0.67.dist-info/METADATA,sha256=vQeBU1Jd6Hc30wSVzIUxBTPnIVTrhpQH1gOAschjqJ0,2982
fjformer-0.0.67.dist-info/RECORD,,
