fjformer/__init__.py,sha256=7EyUxLwbjxfU-_vc0vHYhKFAVW0uddtFmueGsupstNQ,3867
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=EVnVSNEsD8nFwoimvMrvzBVWjDVOlByYsn0b_PnVGxQ,27035
fjformer/attention/flash_attention_tpu.py,sha256=L1xk736VVK84nr3BtVQcJd3XArA5mt6K_SAzxiIE8Lo,51838
fjformer/attention/jax_flash_attn_gpu.py,sha256=T_Weh90o3KxR4vY4eXqXzW3F26cPOwYLiqDZnsi5M8o,20352
fjformer/attention/jax_flash_attn_tpu.py,sha256=wfz_WAaREYOXSpKSjHhH619K30jWgYHoPWl_gcmwBi0,48838
fjformer/bits/__init__.py,sha256=G4h9AdNcsR7UUpnR8kM8WRnJzCV9Svlio8abtrP55Nc,678
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/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=_OXSDA9ahRx9oeoG6Wptboq7rtEFq4z2awDhRIbWLd0,5942
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=fKJhFekLKrYe43HHD6XTC3Haehw_XSTFrZ5AbhQSTnU,11456
fjformer/partition_utils/t5x_partitioning.py,sha256=mxmncTAX3I7NcVxD_bwfZmuDNKmlrHTtCgaH_K1ZiSM,46443
fjformer-0.0.21.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
fjformer-0.0.21.dist-info/METADATA,sha256=ml9sWc13h-GVsaAC4QiFr_tS_mWtI6dfbBIto6kpi1U,2629
fjformer-0.0.21.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
fjformer-0.0.21.dist-info/top_level.txt,sha256=Q09wxLnkDTboxbXuscFCDpgGENPG5JCFyEgrqPzgJC4,9
fjformer-0.0.21.dist-info/RECORD,,
