fjformer/__init__.py,sha256=_aAs66fWsNaBnhs4VbJDrQIe8R1x6y0m68Z3Cy40ME4,1306
fjformer/utils.py,sha256=5anh5HkFeHVaO1yr7wH7paM66vhlNPhhQThX4hQ0e5M,5929
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=Shhk7_sQin1z5BKQOtVFe276ZTfxwmbsQlK7Cvfq9J4,340
fjformer/checkpoint/_load.py,sha256=pBF7nW3lGVE9h10QqFWJDiJd-ACumYdyZxNNm4TdelY,3262
fjformer/checkpoint/streamer.py,sha256=mLaUWGbkyFEkOCiVNqA_342wiFL5iA1PL7s_Dgi29aE,21420
fjformer/core/__init__.py,sha256=boBz8JwZEYTnvFUvLY_WrJ1NM62HrrGHY76bLkuGw5s,1240
fjformer/core/errors.py,sha256=ON9GMmkqnwz4wXDqz0jkUcXVcOtoWqJaGSYSGoX8RMw,1639
fjformer/core/implicit_array.py,sha256=_YtobA9pn0bz5H6sfjVKzai8vP7L0rvaHcaBSfWiJo4,30063
fjformer/core/symbols.py,sha256=RsFgT2UrL_-A63EKvRw0eVPZSMqEDlBOYQ7IkokSvQg,11976
fjformer/core/types.py,sha256=IAz7frAIHxw-q0NubpVRaHq-4Js5qsFriXT53jEfrM4,3684
fjformer/core/utilities.py,sha256=uytwkoqhP5yd7Z1PNE8HfqWMtwL0d4fOVymLrILSLPE,6783
fjformer/custom_array/__init__.py,sha256=YSo5x5v6-Ld0MLg1selduWcGQ5f2TP-Jq_-GHr0EHLI,134
fjformer/custom_array/array4bit.py,sha256=P5tzeYbDvgR6lqXZvKNzIxsR7UaRxDqVUp1SPN9UlOU,21879
fjformer/custom_array/array8bit.py,sha256=vIMv_1OYky0177SIVYDKM1X5BOUKjScjU5h9yWoR7MQ,15524
fjformer/functions/__init__.py,sha256=LCjPVRlujpfCiTcxbuMDanqSujJ8nxwwxPN6UOxQJ2U,1258
fjformer/functions/_functions.py,sha256=3l1_V6jkWsUmfLf3mnJxe7MQ4-O7X8QQE8KHn0FAi3U,2433
fjformer/functions/loss_functions.py,sha256=b8HHxJZjfisFB4xX2nZvV7f46HXfPDahhOUmZVNxvBM,22173
fjformer/lora/__init__.py,sha256=saV1FQDPbaGuDVFO5XNVTeqbfFVcmDel3ccAyNWvVoA,172
fjformer/lora/lora_core.py,sha256=Gm9soQNpB-XhmvyjoNNrOX52Oj56BKCCotvXRopeUg8,6801
fjformer/lora/rapture.py,sha256=mrGj3kJoUCrOfKT5kSBdn3gAa7_6D12PSObivNat220,19605
fjformer/monitor/__init__.py,sha256=k3j-Gk2ZFAnOAOqi-R1qbabFopPMnZPdP-GLkvGf3gc,392
fjformer/monitor/tracker.py,sha256=ghdx0KCJjRmVEIDy3yEID43Fx-ewcoJtrP-vdN198H8,6642
fjformer/optimizers/__init__.py,sha256=yAAJdznQtL55IInROGdQU3EHyPAN7PWAolt6y1uriSI,1605
fjformer/optimizers/adafactor.py,sha256=-BLsrz1rDx-cBk_nHhObGCXSsdgj6bh5bGk-e4xU5Mg,14785
fjformer/optimizers/adamw.py,sha256=qQFh-K2nmkreUefJiDUhaVu_0MkAW5nXj5DTblBRZDE,9806
fjformer/optimizers/lion.py,sha256=UNyb_xB1o4fMYoR5YFEgxVgaoVyezCa-X5yQa6HCLAM,9045
fjformer/optimizers/optimizer_utils.py,sha256=CCVcLRH3JPFqzKgEaDTcsCdDcv4p9bVE5oY3BdcCYAU,2242
fjformer/optimizers/rmsprop.py,sha256=DX6iXzT4-3fKu6RaiYkVh9iaPUQ9KIaDkU085dLfU3E,10272
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=b0oJzebmqXlGqu8oBQ3y-zvOLbvtuk-YGJ_mdMEjtOQ,687
fjformer/sharding/sharding.py,sha256=WTgInuq8xW-4pfSDe8Fdfv6PNLBHFOaviyoBhG52UGE,14322
fjformer/sharding/t5x_partitioning.py,sha256=1tQYv1fyI_hQqqYKdVUbBmgQEetY5EW0a1YumQB-FgM,46052
fjformer-0.0.72.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
fjformer-0.0.72.dist-info/METADATA,sha256=D5r7wdM8tYtEMXaCxd3RvxLhR8Kh8IB90Swn4kfky9o,5057
fjformer-0.0.72.dist-info/RECORD,,
