fjformer/__init__.py,sha256=Lx4ZRgGNMpa03cjT5L9cSjexYOUO1N3SrcuhBghJV80,3857
fjformer/utils.py,sha256=9TnqTlgjEs751Xmnjfs7u2oK0H6Pxi9Ry816IFuE2Nc,5161
fjformer/bits/__init__.py,sha256=WK3qoazE7RYX9EEiXfRq6HGA0AdMv60jlCgFBYBldCg,684
fjformer/bits/bits.py,sha256=LE6phQhZzOV8y7Ng1yO0FvYeGSj-U7PqLRReMSRrHWU,5059
fjformer/bits/calibration.py,sha256=B1afi1oY0D5FMIyx1PKzmo5Tm0A6L2jfxPcQCU3xQ7Q,1748
fjformer/bits/config.py,sha256=BXlkJd0TupfQzPWgb8MaxsAFU210hbfZXtI6DKxnaX4,17587
fjformer/bits/int_numerics.py,sha256=Fnop788DNxkTxlDXn3sfYlGKbiww7e_CEBo3JSSC98c,3827
fjformer/bits/no_numerics.py,sha256=Cxx7Oxm5D1chuc3GoSy8q2Al-KEKNx94nUXNLqIgUbU,1466
fjformer/bits/numerics.py,sha256=v2XLGnxR4LOCkGyG16vZxonNqjmCYS77eatJIHZi1Xg,1538
fjformer/bits/q_dot_general.py,sha256=tzGvsSjZq93Qf-pdSNYdxdLc-6WaG-uO5zMwxuFer30,22994
fjformer/bits/q_flax.py,sha256=ekR1rdK8snpAquql4lJ-XgXQSOcrAjBcFufuYQLA62s,14808
fjformer/bits/stochastic_rounding.py,sha256=qDiulQTPgg4eBrPWvWdN64qJQkjlzvsjzOueK-LBXgA,1495
fjformer/checkpoint/__init__.py,sha256=GcrsyC2r_pquvnuKcNqo2XG6jjM0a0VH3GHDS8j1Qmc,176
fjformer/checkpoint/_load.py,sha256=eD4Tiaz8Go74Pbbk5M3XW4CZUOTGruey2eopJeKgFuI,3235
fjformer/checkpoint/streamer.py,sha256=2z2_l6TcvxZP6Ti-UxuvoZEaSgxqBHDy5Zzx43UKSnA,13678
fjformer/datasets/__init__.py,sha256=FVKH-Q9s1df3SLNXwxjyi-NdVjI-uk2axGyGdJw3YmI,37
fjformer/datasets/datasets.py,sha256=fFK9_0PFJZIjb3-FTDFkt2RuAyIBFHQq3rjUVNrs5Es,2154
fjformer/func/__init__.py,sha256=Mq69rK59o41yBBoScdta0Pb-S2ikC9YfTRuPd-V-J0E,74
fjformer/func/_func.py,sha256=jpJD5WWBECg69r-w6D-6NZbEJ1FyVTjZPrj-dwsbY-s,2020
fjformer/func/loss_func.py,sha256=BrH2liCGSwz0i_0RkAQM1rIbQy9SqCA4_KX7z0UqOtM,21249
fjformer/monitor/__init__.py,sha256=SlAJBaEDyCr8ZMxesPH_phIxr7L_J3OhMWKw1m8f7nI,256
fjformer/monitor/tracker.py,sha256=85zqD47o_W-d8cqQuaNdVXJpxXbynl0KrMPsD4xOu_s,6920
fjformer/optimizers/__init__.py,sha256=xPNOMx2Atup1jKmLYub_1TFRMIVu5cmL3_0lfObt8io,834
fjformer/optimizers/adafactor.py,sha256=DJnDvJnkzs-sWuYyErKk9HbeMiBe59AWZdB7IybtITw,8759
fjformer/optimizers/adamw.py,sha256=L3dfJ5QJspVuFchgo20Sa91r6rd6TKxLZ6biAFGsOTQ,5847
fjformer/optimizers/lion.py,sha256=mVHU6leJ0sk1lZFiaQeNqzU-gUtCsJYZbsPrj4U0IGg,5909
fjformer/optimizers/optimizer_utils.py,sha256=trI8jIUeUc-z1SxX0DIOL1Bw2yDYezQBMpZquIFr8eM,1064
fjformer/optimizers/rmsprop.py,sha256=PmhbWHFmfwAWkrVpNczs3uwRBEasSygvEnTwqiH9suE,10120
fjformer/pallas_operations/__init__.py,sha256=kDf4jL5GZEW9mBcWNs81G7qzfyMPMxZj748HHrMy4LY,759
fjformer/pallas_operations/efficient_attention/__init__.py,sha256=2n4e1R5ab0LG-gxMJiP3E3NVprs_tSIySYpucO1Oqwk,53
fjformer/pallas_operations/efficient_attention/efficient_attention.py,sha256=zV9Phfv-frTXveYUq7Uirepr-DFivmj0JoiN5bgptPA,8042
fjformer/pallas_operations/flash_attention/__init__.py,sha256=Ej1ydthhG7KU6aV0EcRsybc1uMEbyqJrbwuOQM2l7aM,88
fjformer/pallas_operations/flash_attention/gpu/__init__.py,sha256=chRM3UT4XGObE0dV89DujJuFHUjWyxhSWcz4vgqjYag,37
fjformer/pallas_operations/flash_attention/gpu/jax_flash_attn_gpu.py,sha256=TUNx3cP2Jn0nG6LzhgjF5u6J-H7fp6hnHZbpOD8m6Sw,20572
fjformer/pallas_operations/flash_attention/tpu/__init__.py,sha256=DXBwHIB1PsLC0APknr9fUAfio-XYM1BZx-ORTMgxdn8,72
fjformer/pallas_operations/flash_attention/tpu/jax_flash_attn_tpu.py,sha256=hkOb6JI7sriRcLtDvsNRmFwU_2jc-dZGA7jejKHvzxQ,55098
fjformer/pallas_operations/layer_norm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
fjformer/pallas_operations/layer_norm/gpu/__init__.py,sha256=5R6eY4P6QTGYkN-GghoMIQjZ14wexEq-dLJEppBmJ2s,35
fjformer/pallas_operations/layer_norm/gpu/layer_norm.py,sha256=wTjmQ2-fSiH_BzFn2-Np6gmodN1XmkGbim1t7qF1--4,11606
fjformer/pallas_operations/rms_norm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
fjformer/pallas_operations/rms_norm/gpu/__init__.py,sha256=cFRjwfWd4fs7-Z5b931iDyDjIEhyMjYx-FbahFafE9w,31
fjformer/pallas_operations/rms_norm/gpu/rms_norm.py,sha256=4v_CHmNXI0dyVyqr45c2gBZ8gxxwULRwRVT0Lv3ulKA,10434
fjformer/pallas_operations/softmax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
fjformer/pallas_operations/softmax/gpu/__init__.py,sha256=83Z7CzYDHy0RIEeisCPFcFCYY0SHRX4WH_GhGxd__FU,29
fjformer/pallas_operations/softmax/gpu/softmax.py,sha256=mCASWrduaNr_viDXoENkHxLs8GhARWsdVkcLRKVYLe4,2765
fjformer/pallas_operations/splash_attention/__init__.py,sha256=4EWHHTHxwciAWwZeLu5HJ3liVkF3geFayxSxTrEb8GU,648
fjformer/pallas_operations/splash_attention/tpu/__init__.py,sha256=Rtw_WktBe68oSBHkxsTfoF_Y1rlfFkMMiiZSANqyXZc,750
fjformer/pallas_operations/splash_attention/tpu/splash_attention_kernel.py,sha256=_Yd8LGPs__ynuTXRaGgZVh2s1tPadFlF0WwB63u-P-A,82508
fjformer/pallas_operations/splash_attention/tpu/splash_attention_mask.py,sha256=rmLFjY8K1X4NnW3ujdutzYewAH3PzqkQ8XGR9G-geVU,16350
fjformer/pallas_operations/splash_attention/tpu/splash_attention_mask_info.py,sha256=9HWuqtRvvhqiFXDfMprxWpA0MNomX9MUnpfwC8-fJXw,37264
fjformer/partition_utils/__init__.py,sha256=vvYKV9F2n6vKMRfBYb5UEMRt5hM2Hqopb02F1qnHZos,363
fjformer/partition_utils/mesh_utils.py,sha256=fKJhFekLKrYe43HHD6XTC3Haehw_XSTFrZ5AbhQSTnU,11456
fjformer/partition_utils/t5x_partitioning.py,sha256=mxmncTAX3I7NcVxD_bwfZmuDNKmlrHTtCgaH_K1ZiSM,46443
fjformer/xrapture/__init__.py,sha256=4Sxpk8mrsYUTEs9NqQj3FcBybNklc3G_QcJA5hZ7iZM,439
fjformer/xrapture/implicit_array.py,sha256=UtRUdtH6-H5FxrxRXLsXVDWHI1lkj0IuuTvlyaNjnhc,37106
fjformer/xrapture/tracer.py,sha256=hMB3sUoq6BizW17-JH0-ezTr7eY9yn9u4LjHpMmUm5s,2920
fjformer/xrapture/xrapture.py,sha256=5GIoJUMNhA4NkTn_OQFTPxfxqLCIjzo9yjp6nWPym_0,23536
FJFormer-0.0.30.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
FJFormer-0.0.30.dist-info/METADATA,sha256=_wU6qXcGP4K5pim3MdA4EVEOijf5CITBayNrOQ26q0Y,3207
FJFormer-0.0.30.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
FJFormer-0.0.30.dist-info/top_level.txt,sha256=Q09wxLnkDTboxbXuscFCDpgGENPG5JCFyEgrqPzgJC4,9
FJFormer-0.0.30.dist-info/RECORD,,
