fjformer/__init__.py,sha256=lErXLIeby4IeBgP9xMMrlhPlVI9yEWFJNvYu0i7E99Q,1584
fjformer/utils.py,sha256=yL8rktHmZrFOdhQ4V893KBRwhMGYz-OmxeJoiV3yPbY,5346
fjformer/bit_quantization/__init__.py,sha256=knCtl-rqQO95u174QdHgH3sZFwxCfP1rOS9r2M7VxSU,1003
fjformer/bit_quantization/calibration.py,sha256=yjUjtMKd5FX9JlbgB1z04JPzTAYEbpbHcylKTyn_tOw,1651
fjformer/bit_quantization/config.py,sha256=rKS1UdFtSub7Opj0uLmM-_M5Y_hxyF8oXoZjrRaBX1E,16100
fjformer/bit_quantization/int_numerics.py,sha256=xif3d98m62X7MLSsMKaeyHiFj4Jz4IrX_VV3_Fw5l58,3353
fjformer/bit_quantization/no_numerics.py,sha256=bu_ddweoc4LmK8dB4LjGRZ9p9rGxTGEiYwAwjeFUR9w,1294
fjformer/bit_quantization/numerics.py,sha256=XztGQbX2ii72VVGWV0H44f32pnyGT_TmtnuWgxU6R20,1356
fjformer/bit_quantization/q_dot_general.py,sha256=V8qdTApxGcApxvWjO7jw9yAxW3gxLct8Wfp_xEzdQnA,18271
fjformer/bit_quantization/q_flax.py,sha256=YNJki4ze0Yz1aJ3bvNRdFvh403ckXe4wVDq0pJIJ3OY,12677
fjformer/bit_quantization/stochastic_rounding.py,sha256=iB4sQULQRa5IHKea84iiqVRT9yd0cxH-gpEyLer_Fpg,1440
fjformer/checkpoint/__init__.py,sha256=QLs2dI2MEXPVNsxmEDuoj53sK-SEUCuwgpf-YVukrcs,325
fjformer/checkpoint/_load.py,sha256=cwX8xhn4a6eWB4kwScTAFAvKOoJdDTA0dJlVdC1huS8,2818
fjformer/checkpoint/streamer.py,sha256=tZowAIj0b4H5dZnx-HxOdacvPjWGBKIWuE83erie9Ck,17223
fjformer/core/__init__.py,sha256=B15VyW9auIBV3fz7Zwvq0FytLF3lv4tFWXUKXyZlq-U,1303
fjformer/core/errors.py,sha256=dAWzNe6Guu1crY4JdXHO0iPeYuFhLxKOedS1Q7leaew,1483
fjformer/core/implicit_array.py,sha256=JoNwyZ5Hxm1HrbpQJag0FuEif_G_xR5bMy-ADEqSGZM,26593
fjformer/core/symbols.py,sha256=9fNmtryxQ6xPiw_UAwPVmakeU6l9Vgvp9jHprChGc04,10860
fjformer/core/types.py,sha256=ixIVb3zQWNgCMgyQsatFXnSKq1KQWFqFUhIKBylsPGM,3213
fjformer/core/utilities.py,sha256=Yl5t5nZdglacUxYfozsdI23va8NUYJMPmBDoo4DqkEY,5880
fjformer/dtypes/__init__.py,sha256=VpHsyHyWMyVfGS9kzyIXkGCc4IUPYFEiZeM4VujenIU,248
fjformer/dtypes/array8bit/__init__.py,sha256=hqueLcwAI1QIns6EZxmj6crxSEoK93v_rTNb1sz3Y6A,83
fjformer/dtypes/array8bit/array8bit.py,sha256=_A99ltm8-lVXdVFWbhLkanzZe7u_HNJoRrkjGaIFOwM,19687
fjformer/dtypes/array8lt/__init__.py,sha256=CBqhxwdM_1ZOV5XZ40Kfv9n8M9pgHxQRfqn2EVO-REc,79
fjformer/dtypes/array8lt/array8lt.py,sha256=1c-M7RUfvHNlSOxwrM0StTsWOj7uVJy9LV9NMJmK4VA,16650
fjformer/dtypes/arraynf4/__init__.py,sha256=80tL6dDj38wVk-_Rd6lY3KmHOHwuA1AI2ufEb9UrLd8,79
fjformer/dtypes/arraynf4/arraynf4.py,sha256=d20IdCAIzcetWJbWpGw91ERBWWJQ_mWft30-uhWPGqM,17414
fjformer/functions/__init__.py,sha256=rVVwP5r0EnPtcQkMG9MC4Fo60AvZDfX5RduujMOvjQs,1297
fjformer/functions/_functions.py,sha256=kE58w6gne6GrDAYilklTH6HOOC1nsOG2F1KwlpeJCZc,2437
fjformer/functions/loss_functions.py,sha256=x_bhdBCKqTPDqL9H6xXnpp4LyVhK6kcI12WZskNFcZE,20217
fjformer/jax_triton/__init__.py,sha256=ne763Ssb0QVAh1zYZPfN-onVU-_swyCUVOzYWBrURAo,1509
fjformer/jax_triton/lib.py,sha256=eHacjpOgb7puFNeeTP0gKhMTpZ2ndUcj7u0jeNfkKIo,24051
fjformer/jaxpruner/__init__.py,sha256=Yqi_t2Zqe0Sd39YbQ_EjrgxlO2O9Ml-Lf6cTkjOzVqs,2708
fjformer/jaxpruner/api.py,sha256=S6XfQAFQcNtp6Um7jCJ6ab9StjtCBVwDgBhE3Sxly1M,6605
fjformer/jaxpruner/api_test.py,sha256=ADZsqwuNgnJLIv-LgAYf-iHIL60gHkbwfPPiaNuj_S8,3635
fjformer/jaxpruner/base_updater.py,sha256=rFrbab5aeAuyKEdabFlHmp31SSqOOECxvUp6lvVdkt0,10509
fjformer/jaxpruner/base_updater_test.py,sha256=4m1AU_3uC-OdqamRm8Q2zzu4ZwVPsueMn7h1KtQDvD8,4982
fjformer/jaxpruner/mask_calculator.py,sha256=PSz1NsCYCkuyWZ2QHKgZZJXa0Wq2olvoDVwfuodOfu4,9453
fjformer/jaxpruner/mask_calculator_test.py,sha256=kQs-uC4dEyLSjAEFeNiiwxIVDVBa4y6ynxBh1rPhT30,9559
fjformer/jaxpruner/sparsity_distributions.py,sha256=-NgLacRI8Uls5N5D7N91vKM1G8nLJglkFnerrH-iziU,10175
fjformer/jaxpruner/sparsity_distributions_test.py,sha256=X1NHggF2pZH8D4Xoa5ERNj4PHFUE6JyCud2dTOjM7Ms,5554
fjformer/jaxpruner/sparsity_schedules.py,sha256=CdseOU1xJ12VP1YBDy1MN99eHZqSQIoYk90EQMs7AFk,3656
fjformer/jaxpruner/sparsity_schedules_test.py,sha256=u3nGh_2DCSRjCivaAkIFFTQqZAFnBB5Z06OrVUbw4pc,2981
fjformer/jaxpruner/sparsity_types.py,sha256=nrnD2d1UcL20-B-l5RHEGZxx1nsHGfy0mjSDeWaoYMY,1894
fjformer/jaxpruner/utils.py,sha256=WzvQd_1-FU17n_zJTFao6tU3gdZ6XBQ3HOM6Ocz2TA4,3486
fjformer/jaxpruner/utils_test.py,sha256=VcSjAwEliij-JMscaASAHO59euNCtb6eyid5ODM2fqs,2323
fjformer/jaxpruner/algorithms/__init__.py,sha256=nuWIeKXKqdKIbbbQWqUXce50Cog_6JPaao_RgbXkAqk,1534
fjformer/jaxpruner/algorithms/global_pruners.py,sha256=Jw-y-ENhzsGejDJl7kmvW4RLrAN8ERDkyn-e5Z5dJWc,3993
fjformer/jaxpruner/algorithms/global_pruners_test.py,sha256=w1cJM03kuIdLUmSI1hWcmp3JNRouim_kXzSgdPanKyQ,1677
fjformer/jaxpruner/algorithms/pruners.py,sha256=fkrzbcdA9qlt49p21sJQ5SHfxPRL-zYvvRbFv4RHhaI,2377
fjformer/jaxpruner/algorithms/pruners_test.py,sha256=W8srbOPAEyj3mkGmp2OlxEF4mPjFNcgvGEGoIyrG05Y,2259
fjformer/jaxpruner/algorithms/sparse_trainers.py,sha256=Tx_7UtFBkmVzVGLxVeq2wSVCpvB4_YuxN2fr0D-FD5U,6079
fjformer/jaxpruner/algorithms/sparse_trainers_test.py,sha256=n1otKn6wkzlI-OTG-rh8fmfDAWuGyx92Zt1wOH3jooo,5373
fjformer/jaxpruner/algorithms/ste.py,sha256=0mIV8c_K6jcZgiD0vEbe36nWRQyOVieVKdXyHj4wwnY,1988
fjformer/jaxpruner/algorithms/ste_test.py,sha256=4RSutbNS1nxJisz9z13FgJU6ha0H2PVHfHTndw7WJD0,3880
fjformer/kernels/__init__.py,sha256=Jct_Ae9H6X1tGVOSXtik66cNZ0C_ltlhHrms4zdMF6Q,642
fjformer/kernels/flash_attention_2.py,sha256=2kI6tQ1Xm67EjTX0mz-3C3s-v_wwRvcWf0k9OCxM2Z4,5839
fjformer/kernels/gemm.py,sha256=0t-DtFo3eVmMorWqPd2Q9BESQ5PfhsuVl0IWZXUlcDQ,13214
fjformer/kernels/cpu_ops/__init__.py,sha256=7u-EfY5_aaxpgtlVQ1eRUw3Mf2Bg4BSIIA534zeEajY,696
fjformer/kernels/cpu_ops/jax_flash_attention_2.py,sha256=ON3iCEeDTE-_62OfsHohL-TYPxTFublFBxFBAJlQGUY,15279
fjformer/kernels/gpu_ops/__init__.py,sha256=-wzEAU_xBxdADtu9GKDfX52LcAbocoJGHowv1jXQS3E,975
fjformer/kernels/gpu_ops/pallas_flash_attention_2.py,sha256=xT5Jr0IqXcfe81cP-BXvA6eO3qOnLh99X75FXTYghg8,17970
fjformer/kernels/gpu_ops/pallas_gemm.py,sha256=BVHbuhxAAJIRhj0FOaPdjMl_2chYRt4OuzS-Ggrow7w,7348
fjformer/kernels/gpu_ops/triton_flash_attention_2.py,sha256=aKcgLhq3eov1bwKD-ocXWb8QFDjsVF9y-hAuFy_zF2k,21209
fjformer/kernels/gpu_ops/triton_gemm.py,sha256=jDMeSOBXeVY6gqvILAvCwgaiD7CzNJWJd69ai66tGVg,12371
fjformer/kernels/tpu_ops/__init__.py,sha256=OjQ88ZuOemxtUPbVn_iwuHa8VRLRGRagBH4TPcGhQM8,750
fjformer/kernels/tpu_ops/pallas_gemm.py,sha256=i3HRzxWhhWLX8A1s9J4JyG7TvWUGP-wJeLsjrIFxMas,5438
fjformer/lora/__init__.py,sha256=gmMjfaaWZBamiu0v18iH87HTbjZWkUCRQpuqRwNGIoo,250
fjformer/lora/lora_core.py,sha256=zYZvEdluDHoMaJtTszmIqUxz2NKevzR3RcBA5PuB4SQ,6239
fjformer/lora/rapture.py,sha256=rNqtxGfRA45zk5YI-U3pOWB6cuiMORNRDJJeT_mzEqU,16955
fjformer/monitor/__init__.py,sha256=zOQlp55NoOMsO_NdGtqHdrbvyKQ2cNshsora-XIGsjA,424
fjformer/monitor/tracker.py,sha256=GC05EMPB6THciZHloHRgK7xdi9dmqTnmkXCzqa2tj3c,5779
fjformer/optimizers/__init__.py,sha256=GvjfC5GRN8873SPuKvzqqM54vNpcFOQF55uQmv1NU0s,1554
fjformer/optimizers/adafactor.py,sha256=NePfF0qygZxTDqh2-X1CfvJUShXe8SUjGE1T_h0Y_uo,13576
fjformer/optimizers/adamw.py,sha256=rOzuf0GHrPcVqLTYUdeoZQ8gaSLLgtDlcGftE-XR_7A,8929
fjformer/optimizers/lion.py,sha256=bOtcedytgO2EYVeCVU4vSMDX-ede2vhFjbxgOwE1v6c,8312
fjformer/optimizers/optimizer_utils.py,sha256=vA-3acHdDqNzw1kONuUcczX1XPOINcTCsNYmtfauEwo,2028
fjformer/optimizers/rmsprop.py,sha256=tKyKhStva-gVDEjN3pMX4q8YeMH2e4D_1jgcXjcMUHY,9086
fjformer/sharding/__init__.py,sha256=eIoXOQNr2o6OPeGHKNDHF2MtPxAX3Km5_XFsJu82QyI,1202
fjformer/sharding/sharding.py,sha256=YCAmaV3k-AVb7_Cm73L7MePcV4O2aroIPD__09LLrnA,17239
fjformer/sharding/t5x_partitioning.py,sha256=MlOp62Rl3S2fiGQtCHDhUXg0csIm2sXFKhMDNzG94p4,40420
fjformer-0.0.83.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
fjformer-0.0.83.dist-info/METADATA,sha256=4gltUEJN4NNFfn-DCj8gbX1uj3s8iN_CVwF_ebaZmr0,5071
fjformer-0.0.83.dist-info/RECORD,,
