| Path |
Digest |
Size |
| fjformer/__init__.py |
sha256=zJAmtxRaS3wX_Uc0uEw_PxlwFQPoWmMMHUpaHCR7mro
|
1596 |
| fjformer/checkpoint/__init__.py |
sha256=QLs2dI2MEXPVNsxmEDuoj53sK-SEUCuwgpf-YVukrcs
|
325 |
| fjformer/checkpoint/_load.py |
sha256=cwX8xhn4a6eWB4kwScTAFAvKOoJdDTA0dJlVdC1huS8
|
2818 |
| fjformer/checkpoint/streamer.py |
sha256=pR_Tb2RLeivwbMIR0hMYPs5tGDqnVdbJUTgSg4-OhpM
|
17434 |
| 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=gegn4zJEJ5UKkLh8hRwhcdWdLqwQdE63iPu0OcNp40U
|
334 |
| fjformer/dtypes/a4q/__init__.py |
sha256=4ozrvinys-aHhVzOlarMRw958bA1rXmk-Ds8yriMs5Y
|
59 |
| fjformer/dtypes/a4q/a4q.py |
sha256=3UI1oZlth8Q9NaPMLsYHdoqJcHUbH1BCibQ0Mave1hQ
|
13037 |
| fjformer/dtypes/a8q/__init__.py |
sha256=rg2jS4kdY70VviXOqzZh0xxXwPdsfNLpWOHClmMxKYY
|
59 |
| fjformer/dtypes/a8q/a8q.py |
sha256=meNM0aGz6tcxwiXK8l6d49H-jXEF9z4SCdPPFP7yYm8
|
13041 |
| fjformer/dtypes/array8bit/__init__.py |
sha256=hqueLcwAI1QIns6EZxmj6crxSEoK93v_rTNb1sz3Y6A
|
83 |
| fjformer/dtypes/array8bit/array8bit.py |
sha256=v8zqmti-OeAqGHHNT2upLwAePirMLddn0rKa0iv-YAM
|
19781 |
| 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=NKzhxqNsERzlHsALcbbCHsKAeNOtyZv7ruMpD_rXdyc
|
16895 |
| 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=qxDaBT_CjI30hzThpaJDa4lL-OBwfZyllSVElLz32ak
|
1382 |
| fjformer/jax_triton/lib.py |
sha256=dBY87U1Dil3Rwr8aodPjbB0U9qlMR2smP_lYptmLvM4
|
21998 |
| fjformer/jaxpruner/__init__.py |
sha256=Yqi_t2Zqe0Sd39YbQ_EjrgxlO2O9Ml-Lf6cTkjOzVqs
|
2708 |
| 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/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/kernels/__init__.py |
sha256=Jct_Ae9H6X1tGVOSXtik66cNZ0C_ltlhHrms4zdMF6Q
|
642 |
| 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/flash_attention_2.py |
sha256=2kI6tQ1Xm67EjTX0mz-3C3s-v_wwRvcWf0k9OCxM2Z4
|
5839 |
| fjformer/kernels/gemm.py |
sha256=0t-DtFo3eVmMorWqPd2Q9BESQ5PfhsuVl0IWZXUlcDQ
|
13214 |
| 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=Jpqh9ygKNPRFs4dsmg6WB_DCqee8fXgv2ln9zuiJnlc
|
20105 |
| fjformer/sharding/t5x_partitioning.py |
sha256=MlOp62Rl3S2fiGQtCHDhUXg0csIm2sXFKhMDNzG94p4
|
40420 |
| fjformer/utils.py |
sha256=1iikvI-lryB-1WwENJfdvRVDMi9PVjJ77601lGUBVLc
|
5359 |
| fjformer-0.0.91.dist-info/LICENSE |
sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ
|
11357 |
| fjformer-0.0.91.dist-info/METADATA |
sha256=yuT-E4DjaXIrxT5MIF16miu3lxXga_q48dpqxhBoGTk
|
5203 |
| fjformer-0.0.91.dist-info/WHEEL |
sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs
|
88 |
| fjformer-0.0.91.dist-info/RECORD |
— |
— |