fjformer/__init__.py,sha256=yVXeyq62ZlX4vgBhCGe8V6sQRyuAc3h5iGa2I5e0xEc,4181
fjformer/utils.py,sha256=9TnqTlgjEs751Xmnjfs7u2oK0H6Pxi9Ry816IFuE2Nc,5161
fjformer/bits/__init__.py,sha256=nSV_2nNED6I_bv6f6jPwJY7fVwCrZrHjAh5ZzambJPI,1310
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/qk.py,sha256=BSf0HHV_4clXEAm4tHEVXrBPWAW24JCkaioiA0ojIuU,458
fjformer/bits/stochastic_rounding.py,sha256=qDiulQTPgg4eBrPWvWdN64qJQkjlzvsjzOueK-LBXgA,1495
fjformer/checkpoint/__init__.py,sha256=ErCZGbje7uCNChvy5dUoNyrDJi8GjEgkqXh8smEMv8A,339
fjformer/checkpoint/_load.py,sha256=eD4Tiaz8Go74Pbbk5M3XW4CZUOTGruey2eopJeKgFuI,3235
fjformer/checkpoint/streamer.py,sha256=2z2_l6TcvxZP6Ti-UxuvoZEaSgxqBHDy5Zzx43UKSnA,13678
fjformer/datasets/__init__.py,sha256=Yo2wx4Sso-fjkXM69K0XbriijJ-o1ds8-xWW1ydrb6w,66
fjformer/datasets/datasets.py,sha256=fFK9_0PFJZIjb3-FTDFkt2RuAyIBFHQq3rjUVNrs5Es,2154
fjformer/func/__init__.py,sha256=_bxsmv_67j2xYEPt97jLmXLD0wSSAIOM2dDb9EwxtCI,1792
fjformer/func/_func.py,sha256=jpJD5WWBECg69r-w6D-6NZbEJ1FyVTjZPrj-dwsbY-s,2020
fjformer/func/loss_func.py,sha256=tZBNIYpG0eVSNIMtNRCyRs6TwZstPFqqSb6M0cpql8I,25751
fjformer/monitor/__init__.py,sha256=tl6_H3ot_khoI6w2CdNEdPM3yGPy7yWZoFdKnxZ-jIQ,376
fjformer/monitor/tracker.py,sha256=D7nXWhCYiYgB5f93SDtI1WTBS3u0uyCD8Ep2R33AseI,6920
fjformer/optimizers/__init__.py,sha256=CxQczf0eKQ53IyQrUlrgIcblQ7O5bvE3Z3G4GsszDcg,2364
fjformer/optimizers/adafactor.py,sha256=zcwAHex0jAW3XEiAvyd_WTKo52wxX_Csfhh1tGAPNi0,8887
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=NajMepV8ZS29GY5qUP5DyKcBF17kPKNgmjsW2WGJnUc,2478
fjformer/pallas_operations/efficient_attention/__init__.py,sha256=MH9HzNgO7yKoDP9zCp3wGhncB6NO0_e15i8v1pOTWnw,87
fjformer/pallas_operations/efficient_attention/efficient_attention.py,sha256=zV9Phfv-frTXveYUq7Uirepr-DFivmj0JoiN5bgptPA,8042
fjformer/pallas_operations/flash_attention/__init__.py,sha256=excKMhByr-e5JgbvuM8-X3UntXdZDo3E2EGFaDieck0,166
fjformer/pallas_operations/flash_attention/gpu/__init__.py,sha256=wwqKQjWJnCga5eB5Su1okKgn-_dIaWc1Pv5DgRbdHvQ,57
fjformer/pallas_operations/flash_attention/gpu/jax_flash_attn_gpu.py,sha256=TUNx3cP2Jn0nG6LzhgjF5u6J-H7fp6hnHZbpOD8m6Sw,20572
fjformer/pallas_operations/flash_attention/tpu/__init__.py,sha256=GQT9yNmvBD7hS4gXVsD9PnmcIY4ZDYv7GxVveKd2BJk,123
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/ring_attention/__init__.py,sha256=UMyZbTTFq8Z9_vNiBXgjoV-DFwrCRgyZmMAjdsHccE0,268
fjformer/pallas_operations/ring_attention/ring_attention.py,sha256=zwb6IW5ENWktLJp9hqBwsgJ1sGnrRBAj9oDRAizwJRA,74781
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=jEuBy5kr94DdL5MJyaJhRgct8XMnQWwoUaZgwJz7uWw,1763
fjformer/pallas_operations/splash_attention/tpu/__init__.py,sha256=v9vyrP6FGq11ubfYWd-xyat03WQ7VKViVGZtojqLbA0,1867
fjformer/pallas_operations/splash_attention/tpu/splash_attention_kernel.py,sha256=6FHzaDvYPXj5JNs5_k9em-wSQh5QG0ZKK8iLaVjLi2E,82553
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=J1H6OydxCJcuCth9xybbrbvCbrOCktaJs-wdc_a0158,34278
fjformer/partition_utils/__init__.py,sha256=vvYKV9F2n6vKMRfBYb5UEMRt5hM2Hqopb02F1qnHZos,363
fjformer/partition_utils/mesh_utils.py,sha256=D2bqCJQ1Ed16gQWhSxCsnHor_RlXNcolN5VHTIDsc50,11475
fjformer/partition_utils/t5x_partitioning.py,sha256=mxmncTAX3I7NcVxD_bwfZmuDNKmlrHTtCgaH_K1ZiSM,46443
fjformer/xrapture/__init__.py,sha256=5NPeL50SZGZg1P0h_weNp0OCjun65anBKsQPUpRJlNQ,1048
fjformer/xrapture/implicit_array.py,sha256=Mj0PXlIuABfbv40tmousJYWmJDfe3O1yPwxAwDKEqlg,37113
fjformer/xrapture/tracer.py,sha256=hMB3sUoq6BizW17-JH0-ezTr7eY9yn9u4LjHpMmUm5s,2920
fjformer/xrapture/xrapture.py,sha256=5GIoJUMNhA4NkTn_OQFTPxfxqLCIjzo9yjp6nWPym_0,23536
FJFormer-0.0.40.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
FJFormer-0.0.40.dist-info/METADATA,sha256=yhI4mopUxwoIi695JjGoyc3hd2OW9-pg5WAIY0tkj5Y,3133
FJFormer-0.0.40.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
FJFormer-0.0.40.dist-info/top_level.txt,sha256=Q09wxLnkDTboxbXuscFCDpgGENPG5JCFyEgrqPzgJC4,9
FJFormer-0.0.40.dist-info/RECORD,,
