fjformer/__init__.py,sha256=vw-toWRJULGpQwGkAARpow-zgjcsE7AtxCF3qJrDChs,3899
fjformer/utils.py,sha256=9TnqTlgjEs751Xmnjfs7u2oK0H6Pxi9Ry816IFuE2Nc,5161
fjformer/attention/__init__.py,sha256=B-5dhh_s7MatgM_KvCltFS4LPePXRkZyEE5yoWeajpw,1523
fjformer/attention/efficient_attention.py,sha256=zV9Phfv-frTXveYUq7Uirepr-DFivmj0JoiN5bgptPA,8042
fjformer/attention/flash_attention.py,sha256=-FhO5HwIDeoCakPW1acwGSvJRNZtx5R5k4lwgldKwHc,45881
fjformer/attention/flash_attention_0.py,sha256=4eYt23ZeRDWB1zBBet6OIurRhRVIQgKojQqrRmeBpOA,25511
fjformer/attention/flash_attention_gpu.py,sha256=bs2nDpScGuPtfw2rWbglXZ-QJpgoS9iWYkGzvdJANrk,27240
fjformer/attention/flash_attention_tpu.py,sha256=Dz2H5Kew3YOXvDUdE9-CQhKYRx1hlyg8Oy31xUnimmc,61319
fjformer/attention/jax_flash_attn_gpu.py,sha256=T_Weh90o3KxR4vY4eXqXzW3F26cPOwYLiqDZnsi5M8o,20352
fjformer/attention/jax_flash_attn_tpu.py,sha256=hPf8gNz1npUFZStnoUJggDVOJi2ncEiLwGjflZS35wQ,60078
fjformer/bits/__init__.py,sha256=5M4rmFwV8V3BUXjwmjh65pMh5XeEcoc3tt_CmJ66Mts,631
fjformer/bits/bits.py,sha256=CCS87DVTl72kqTeL2ehqEFfQDfMVjAHpUhOlgQBH7o4,5061
fjformer/bits/calibration.py,sha256=Mb0m2H7dKCa9RXPERYSg0en26LXxUj8AoKl9oSZWecg,1749
fjformer/bits/config.py,sha256=0btejzHUIcZl1sH97UOFVC0LsM4jXuwmUTzjxQc0HMM,14862
fjformer/bits/int_numerics.py,sha256=soHcSre99fhX_3hwJQG84RFta_Xu15Gkaa8ECeE6Xms,2889
fjformer/bits/q_dot_general.py,sha256=c0Y6tt_Ct8DadaiXpkFEnDT2bgy9e1EYJVz6Ks7mGHQ,17562
fjformer/bits/q_flax.py,sha256=USTnwE1ofs7M1n5GnYFcjBsO4FeRMpDJjKvnc0um5zo,10481
fjformer/bits/stochastic_rounding.py,sha256=qDiulQTPgg4eBrPWvWdN64qJQkjlzvsjzOueK-LBXgA,1495
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=JVPPpwSXD2fe5LQVdlL59J8-XxLCXfJI-dCmxe_BxQU,21233
fjformer/gpu_pallas/__init__.py,sha256=0QgC8PnxRzn6WJijx2pIAvUmbYF7qDLFKD7ioBBdnWE,145
fjformer/gpu_pallas/attention.py,sha256=T_Weh90o3KxR4vY4eXqXzW3F26cPOwYLiqDZnsi5M8o,20352
fjformer/gpu_pallas/layer_norm.py,sha256=dyfN5Qt2yrfd44AhimWzhVEyT7rodZwROaOcwjBHShA,11588
fjformer/gpu_pallas/rms_norm.py,sha256=oLnGsIrDDzD8xGs5QjEMofEz1FV0zbzz_fy-nZaDajo,10416
fjformer/gpu_pallas/softmax.py,sha256=1DN8CrgJU2KZhFKf-VrXp5pbH64Yk-ocxkZmYUOWHDE,2647
fjformer/load/__init__.py,sha256=bD7vszbLcRIgW57OTBwvDVoPwM4YMSutwIpeR3IYl68,192
fjformer/load/_load.py,sha256=yxJqZmPxwkqd24mNuj5CF8NFa_5hyYpVMMg6tjNJJ84,3250
fjformer/load/streamer.py,sha256=1ZKno52S2g-1TY3hiY8k24x6tfzmM2u5Wb-mRp9hVoc,11187
fjformer/monitor/__init__.py,sha256=_weduW1pMMgwVJmKDTeHXNeSa28fMAEN8rdh32gZR5A,82
fjformer/monitor/tracker.py,sha256=GN9lGa0EcywDhckdtDrWcH4anShnaFGvU3yXlih9Ty0,5766
fjformer/optimizers/__init__.py,sha256=z-urwkXawX7sRg6HhcqCyKVZkjgwDmKcuoxBuBQLqXA,646
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/partition_utils/__init__.py,sha256=vvYKV9F2n6vKMRfBYb5UEMRt5hM2Hqopb02F1qnHZos,363
fjformer/partition_utils/mesh_utils.py,sha256=MTGjmt-d0Pbf3eMSCB_-OFAlkxStr5-wRWKWwU8_0E8,11046
fjformer/partition_utils/t5x_partitioning.py,sha256=mxmncTAX3I7NcVxD_bwfZmuDNKmlrHTtCgaH_K1ZiSM,46443
fjformer-0.0.17.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
fjformer-0.0.17.dist-info/METADATA,sha256=Wxk-MG1oNN742NnkzFk5VnIyXGomd0WyEmPCOj8TfwQ,2555
fjformer-0.0.17.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
fjformer-0.0.17.dist-info/top_level.txt,sha256=Q09wxLnkDTboxbXuscFCDpgGENPG5JCFyEgrqPzgJC4,9
fjformer-0.0.17.dist-info/RECORD,,
