fjformer/__init__.py,sha256=sh0zXiTXKBRs9CDA6EUOAYIbqOd7ng9ZHe8Qx7g1ywk,1604
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.py,sha256=0VtOR8D82xV_gCYspmGiSBWqmWA_t72jLedVeJzRKxA,15522
fjformer/dtypes/array8lt.py,sha256=1c-M7RUfvHNlSOxwrM0StTsWOj7uVJy9LV9NMJmK4VA,16650
fjformer/dtypes/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=vSO2Gl9-V5vdOwj7dqmi7t-RpBaGdJICGQPU0WmxRzE,23445
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/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/pallas_operations/__init__.py,sha256=T-brR4y3yHMk3a59uHu1IxcwvvCLTKmfZbHJXZ18wf4,354
fjformer/pallas_operations/jax_flash_attention2.py,sha256=GbGq5dJzlz9E7ObDoPT_HoytvPeD1dzG6VFn0YKKj7w,11209
fjformer/pallas_operations/efficient_attention/__init__.py,sha256=4QnTuUSlYBcppFhRbbBhLzJ0hvtHbXK7f8mt9gGZ4n8,128
fjformer/pallas_operations/efficient_attention/efficient_attention.py,sha256=epbmURSzDaPNQ85PyZjoiy5WYNdkwIc7Fzj_XEhxq2g,8570
fjformer/pallas_operations/gpu/__init__.py,sha256=5hB9CeoI7Abbq-qoMR8pyld1bISkcQ8RTlcCiDkFbuI,196
fjformer/pallas_operations/gpu/flash_attention/__init__.py,sha256=IUFOX-3_w_w0X_xipZPCaX0F29UbzTwtmAs5wqDdbGE,118
fjformer/pallas_operations/gpu/flash_attention/mha.py,sha256=KTTcSHM-axEzMFEHIdqYv6qTDXnuhNdV_FIppnlQ9TY,16306
fjformer/pallas_operations/gpu/layer_norm/__init__.py,sha256=JBLfJgywCFAq_2ptpG4TJ9PMyutOed4O2CUNzHH7eK4,96
fjformer/pallas_operations/gpu/layer_norm/layer_norm.py,sha256=7edOsn8hMQ46wGHls9PW9bRj5zY4M8nUIfqyEM0ZcLI,9709
fjformer/pallas_operations/gpu/rms_norm/__init__.py,sha256=vMZCLLPCJ_QFQ9gSJRzJM4XTW07h5yn-yBQVYDfj2Sg,82
fjformer/pallas_operations/gpu/rms_norm/rms_norm.py,sha256=0vAetho686_Hg4iTJIIRTfhWAGcs-X5gmgpq4CprPzU,8655
fjformer/pallas_operations/gpu/softmax/__init__.py,sha256=JiVkFCpBi42u2NjPdANiKMty_uYTeKRk6J5QBarNmvU,77
fjformer/pallas_operations/gpu/softmax/softmax.py,sha256=6iRUXVpyYiwZ_aS2lNAumViy5gyoeNKRZaFr2DRyuaI,2002
fjformer/pallas_operations/pallas_attention/__init__.py,sha256=MFJ0QIZufYUU1kRAD4L-PnQpE3nv25nXiTnUMr2dw0Y,107
fjformer/pallas_operations/pallas_attention/attention.py,sha256=9q2eY9DDbQy4BCPkKa93o8jI1z_a4k93nu83XABPMvk,13681
fjformer/pallas_operations/tpu/__init__.py,sha256=jFtbEzF3tb5sIuGqGj6D5cTsUVoGB_7VCgJBjeh86i8,312
fjformer/pallas_operations/tpu/flash_attention/__init__.py,sha256=AD-VSL-oRZDNZPRXRc5fhTugzaqhjNBMIw9R8719Nfo,222
fjformer/pallas_operations/tpu/flash_attention/flash_attention.py,sha256=HgsI-BedmF2fFPYOloawpLmDYZVbFVUtueVB7266Xmg,44590
fjformer/pallas_operations/tpu/paged_attention/__init__.py,sha256=_i0BArMSqAKR00526w70knK4GlNA3R_P0dJxvL8ibfc,258
fjformer/pallas_operations/tpu/paged_attention/paged_attention.py,sha256=KsVGxwJ-WMnRvKsCXKjOULpuFIDj4GiFrl6CPp6GNGs,20092
fjformer/pallas_operations/tpu/ring_attention/__init__.py,sha256=HvptyodoMQCkVHh8Xu94Sfm_VLHD7a7Mr5AtrM3jjSU,220
fjformer/pallas_operations/tpu/ring_attention/ring_attention.py,sha256=9vnYq5OAbTiTkJXONQDG2Njr732E2kbOsFSNt4CSNAg,62094
fjformer/pallas_operations/tpu/splash_attention/__init__.py,sha256=mfSrbA4VdMOaYgoMvultyG5wMCRlX-UmwEcDSGnu-fQ,1853
fjformer/pallas_operations/tpu/splash_attention/splash_attention_kernel.py,sha256=O6u6q_0mV5XrGk-LkxjqFkWv9mZPhVqZpRxU5Z6fdC4,67352
fjformer/pallas_operations/tpu/splash_attention/splash_attention_mask.py,sha256=9ug5_VMlnlVlXvhl6bdksAjh6hsL4xiPJywME5JaIqY,14184
fjformer/pallas_operations/tpu/splash_attention/splash_attention_mask_info.py,sha256=L8v_pVN3jbVGJ9A0r0AIBYYfPX9Vi9t8seWBhJ348as,32066
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.81.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
fjformer-0.0.81.dist-info/METADATA,sha256=kJsbV9c_JIDj8rRV9Xg2hGHiegqPIzt8jNUahmoXq84,5071
fjformer-0.0.81.dist-info/RECORD,,
