fjformer/__init__.py,sha256=vSxRR2XxxhEZ4jxEz34-kgeMqX2fLvgXYzG3_DzaSnM,4188
fjformer/utils.py,sha256=3nLxe5H2Kr7wpyVh3DL2Rp-kk5cVchUHw0Ox_KZGCPs,5224
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=y501kjmlqYwDCZlZKPP5eFGUx91K2tAXL94e-pRi5PA,13726
fjformer/functions/__init__.py,sha256=_bxsmv_67j2xYEPt97jLmXLD0wSSAIOM2dDb9EwxtCI,1792
fjformer/functions/_func.py,sha256=jpJD5WWBECg69r-w6D-6NZbEJ1FyVTjZPrj-dwsbY-s,2020
fjformer/functions/loss_func.py,sha256=tZBNIYpG0eVSNIMtNRCyRs6TwZstPFqqSb6M0cpql8I,25751
fjformer/linen/__init__.py,sha256=jfzBt_U8gSastrPuwnXZAIjZddwlu2B94bEt3zNOcHA,1224
fjformer/linen/linen.py,sha256=pJLEvMlM2FGMuLp5bg5UR0Te4Ht_qp1aNaF-co9YDYk,72021
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=wn4rYAemqe0NGLAtObimx3_bkQOB6O_7OKS0WmAbv8c,2577
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/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=x6ET-ZTHUC6yQrJTtlUL80GXIcpk2KXeMxvEy-d1u3Y,11624
fjformer/pallas_operations/pallas_flash_attention/__init__.py,sha256=BzOH4r9TQJYQ2LS6edq6zgaJi53AYtXGidbO1l_rfTI,69
fjformer/pallas_operations/pallas_flash_attention/attention.py,sha256=PwQ45mqP0aQggik8OI7cowzI5tC1BmlbC7wUs0we6zg,16259
fjformer/pallas_operations/ring_attention/__init__.py,sha256=UMyZbTTFq8Z9_vNiBXgjoV-DFwrCRgyZmMAjdsHccE0,268
fjformer/pallas_operations/ring_attention/ring_attention.py,sha256=04Rfnup-SaO3JfrWgKRf2AKmzqS7hiIBNuEDZiHKg2w,74839
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=cpJK6Bbhes6YlIznkvEWUBChwL64VR_gSWs2jtyCEyo,10452
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=1yWOlpd-FNX_pOFwhuw8PbMXTZ0Pumz4VaXM5fH5e1Y,82705
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/pallas_operations/tpu_flash_attention/__init__.py,sha256=excKMhByr-e5JgbvuM8-X3UntXdZDo3E2EGFaDieck0,166
fjformer/pallas_operations/tpu_flash_attention/gpu/__init__.py,sha256=wwqKQjWJnCga5eB5Su1okKgn-_dIaWc1Pv5DgRbdHvQ,57
fjformer/pallas_operations/tpu_flash_attention/gpu/jax_flash_attn_gpu.py,sha256=QAavvrldhOyTQFL6R4-55Qn8RiLPOyQ7dgfCiffJ0sA,20599
fjformer/pallas_operations/tpu_flash_attention/tpu/__init__.py,sha256=GQT9yNmvBD7hS4gXVsD9PnmcIY4ZDYv7GxVveKd2BJk,123
fjformer/pallas_operations/tpu_flash_attention/tpu/jax_flash_attn_tpu.py,sha256=LkjMKiRfkqIsaR0HnkOE49O6uUqlc41ljtBfxuaO9yI,55183
fjformer/sharding/__init__.py,sha256=H_MXn2fHm6bVGBh3V8fJdp0wNmv4k9BUZMbelF_J1FQ,361
fjformer/sharding/sharding.py,sha256=-Xrrj-yVgOOzg338a2-h6p7-XHJ3nN1hzXtNj0auqXo,11381
fjformer/sharding/t5x_partitioning.py,sha256=M-cjOqc8NERAewVomZ7Iwu8y3bUZtzni9qNr0KuZVtk,46455
fjformer/xrapture/__init__.py,sha256=5NPeL50SZGZg1P0h_weNp0OCjun65anBKsQPUpRJlNQ,1048
fjformer/xrapture/implicit_array.py,sha256=RFT9WzIMslbvIcWy87-I8VbIopF76fEQ0xUqxeQkTFo,37136
fjformer/xrapture/tracer.py,sha256=hMB3sUoq6BizW17-JH0-ezTr7eY9yn9u4LjHpMmUm5s,2920
fjformer/xrapture/xrapture.py,sha256=ecQ-yV44fsilRMUhlt3efKh1BH55m3EcfetXIFcuYv8,23595
FJFormer-0.0.63.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
FJFormer-0.0.63.dist-info/METADATA,sha256=EktfpgPPf0BUtne9JPSnScx942LyjVW6AfEv3OWlVgQ,3208
FJFormer-0.0.63.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
FJFormer-0.0.63.dist-info/top_level.txt,sha256=Q09wxLnkDTboxbXuscFCDpgGENPG5JCFyEgrqPzgJC4,9
FJFormer-0.0.63.dist-info/RECORD,,
