jax_linear_attention-0.1.0.dist-info/licenses/LICENSE,sha256=DOk07mN0oL5IoUGUdyRYTDhaqZ9uULJjn4JYpkEN51o,11346
jax_linear_attention-0.1.0.dist-info/licenses/NOTICE,sha256=f2ffTu9NwogBhw1_0Vgtgsm-PaBW3i72wmLfEGyvm9I,20017
jax_linear_attention-0.1.0.dist-info/licenses/licenses/DeepGEMM.LICENSE,sha256=bHOjDBSik4q6pq_pjrsZ5tgdrFvUrODLBcYcIK-je4U,1065
jax_linear_attention-0.1.0.dist-info/licenses/licenses/DeepSeek-V3.2-Exp.LICENSE,sha256=bHOjDBSik4q6pq_pjrsZ5tgdrFvUrODLBcYcIK-je4U,1065
jax_linear_attention-0.1.0.dist-info/licenses/licenses/DeepSeek-V3.LICENSE,sha256=HI9XPoMMqbPr_res4YIxRuIrZvme4iOEDnY3yedF4cc,1064
jax_linear_attention-0.1.0.dist-info/licenses/licenses/FlashKDA.LICENSE,sha256=BfF1BiTWq19t1Z3qeeMVb0_sazBlyUruryZd9nf55ug,1067
jax_linear_attention-0.1.0.dist-info/licenses/licenses/FlashMLA.LICENSE,sha256=bHOjDBSik4q6pq_pjrsZ5tgdrFvUrODLBcYcIK-je4U,1065
jax_linear_attention-0.1.0.dist-info/licenses/licenses/FlashQLA.LICENSE,sha256=0LehEfSNwCB8fgV8gQrg4v28gHumVOtmmt6ZiyzcNDg,1070
jax_linear_attention-0.1.0.dist-info/licenses/licenses/Liger-Kernel.LICENSE,sha256=OhzLDHJ0to4a8sodVLELZiCFylZ1NAAYLs-HrjPy0ag,1312
jax_linear_attention-0.1.0.dist-info/licenses/licenses/MoBA.LICENSE,sha256=8MiYBhVfk_TqXLnoHvcKrRp3KSQBMt-XBtuVhAJFR9s,1083
jax_linear_attention-0.1.0.dist-info/licenses/licenses/RWKV-LM.LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
jax_linear_attention-0.1.0.dist-info/licenses/licenses/TransformerEngine.LICENSE,sha256=vqG0Jm8UmfpLuxbW51d-a1tT0e8BCDWp1GCHYCLMhyI,10142
jax_linear_attention-0.1.0.dist-info/licenses/licenses/cutlass.LICENSE,sha256=gKehi3PUH2TdnKiBrzWTj43oixjHKHAyUfTJTRKZiE0,1854
jax_linear_attention-0.1.0.dist-info/licenses/licenses/flame.LICENSE,sha256=VGB1FpXk7dOdRgoA7iOROJ6SinqtFTjDYEJX2uCuInc,1084
jax_linear_attention-0.1.0.dist-info/licenses/licenses/flash-attention.LICENSE,sha256=jJzLlsBl5wYTW2y60nm3IdphVuUfOl8nxrMymvlBbXM,1558
jax_linear_attention-0.1.0.dist-info/licenses/licenses/flash-linear-attention.LICENSE,sha256=Qag8gYfvwePMwhkJ6Aap5SM45pRIVU2XVHBsPRzWEOc,1096
jax_linear_attention-0.1.0.dist-info/licenses/licenses/flat.LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
jax_linear_attention-0.1.0.dist-info/licenses/licenses/flax.LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
jax_linear_attention-0.1.0.dist-info/licenses/licenses/jax.LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax_linear_attention-0.1.0.dist-info/licenses/licenses/mamba.LICENSE,sha256=dgk5sAAZTQRUjt5qhXu-c10WldhCLshZVcjivX9LlcU,11348
jax_linear_attention-0.1.0.dist-info/licenses/licenses/maxtext.LICENSE,sha256=PWvDbyxiaVFtjMjuY07bL-EkG5UZKAr0_lvdglV5UkE,11347
jax_linear_attention-0.1.0.dist-info/licenses/licenses/ml-cross-entropy.LICENSE,sha256=SGLHf5v4Q_-Pyhka4dEkqUhJEF26WJG8M6qTREBzJgA,2684
jax_linear_attention-0.1.0.dist-info/licenses/licenses/tokamax.LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax_linear_attention-0.1.0.dist-info/licenses/licenses/transformers.LICENSE,sha256=d_1HEN757DwPYiWADgI18VpCWr1KiwNVkSf814JhIEk,11418
jax_linear_attention-0.1.0.dist-info/licenses/licenses/trl.LICENSE,sha256=G_YUsf3ED700SqWNANF9X9DSUuozOgW-AYo4lZyLKZg,11355
jax_linear_attention-0.1.0.dist-info/licenses/licenses/xlstm.LICENSE,sha256=TT8Lao8imQMnCh3PfYsOWaUO7VNRQMTV49B_Rp-b3A8,11364
jla/__init__.py,sha256=HUuIXLviTaMzxJQvnU7jl1poZqi3MkptwOkGhjQrB-c,196
jla/cli.py,sha256=GChq-eOtSWO31O3ZSFtFXbqq1uUXfP0s16fLGqzRcsI,4796
jla/config.py,sha256=qXcnmS6z523DT-HghXqUeu3FZ5O-sSXOHawJfxX5VBg,40236
jla/launch_env.py,sha256=OHz9xaW3Qe9mtnSKTOdBjYNRuDXgC7NQmZQudcCZeEM,4717
jla/convert/__init__.py,sha256=WtIq6BSianjgYg69q0sBNTyYRn5G9-qWizQJREDzYDs,736
jla/convert/cli.py,sha256=wpYuUyfQm0lRcVQY3CuLY8c0ayeb8eNJ3UA8A2RBhPI,6920
jla/convert/hf.py,sha256=sHcG3r0OaosCKUnr2l3CZ3MEjEUQudOWcO04JDd5Ms8,7483
jla/convert/hf_config.py,sha256=-YkB3PiDjP9wjzfVNZpYsKKJYQeaKwNDFb50a2vey7Q,32876
jla/convert/hf_weights.py,sha256=drDB1S3Wfx3APOTrSt0rdBRZSLfLllcFrvrV3uEjrls,26558
jla/convert/safetensors_io.py,sha256=LoPJWJ6rsnfE8bIEogTvBrZnJx0wd6APnSNnSuaNIjo,9364
jla/data/__init__.py,sha256=o2eh1w_ClHzmtjylCSOYDClK91pgggxZnR8h1S19g50,888
jla/data/cidx.py,sha256=jTDJVYz6hawk-WBSBVfn_2JwME4Dl3xxduTgG1AJWvk,12890
jla/data/datasets.py,sha256=uggiyLQC3u_4D7z6jGoogqu0D2Orkxp2DqlN0nwR00I,12234
jla/data/loader.py,sha256=qBff7N1Vk88WBj7oLJrTXab1g3TgOX2_3eFVbmE0OV8,8428
jla/data/make_data.py,sha256=zKHtDWgmePrEWdAKJ4ixdPDSQ5JIwUcvs6nV5D1_fbc,7000
jla/data/packing.py,sha256=tSyaClX_KfJDq-o4RNcmkeafs_FhkD5yTI96L26XyWU,2499
jla/data/pipeline.py,sha256=F0_nxSwnJyke_W5ZyjWI1v8Ws9XoH8y2lemS0zUwsj4,11283
jla/data/preference.py,sha256=OYcxqaguiEGa97PUzL6nJw7QtiqH7VYjGbawUK8GSRc,11103
jla/data/sft.py,sha256=leucOLN1YqD544lQ2WnaIXQAXAjzbDPZQwzwk7r_33A,21033
jla/data/sft_tokenize.py,sha256=M0fRsnP-92E3SUZKZI1b_JEYXTup8Qo2yC9uA9RiYL4,25272
jla/data/streaming.py,sha256=mYS4Zw4u6Uf4sHn-9I6DSwgVeDzY70NNQh5aB5Z81a4,27448
jla/distributed/__init__.py,sha256=9XTY9noM0-C5kvFw9kMQYbETQ01KA7FFdCxn0n0SQKY,860
jla/distributed/activation_sharding.py,sha256=cxNhunOQLCFhUL7lg3c13jLN_wwhn865c1J-Cm2wDuU,3117
jla/distributed/context_parallel.py,sha256=lD-bvKGURTAEEFmtDwUQYFWW6qZW8u2dglmADZoDtiY,7939
jla/distributed/context_parallel_attention.py,sha256=G-YH5rHStFEmkAfjnyZG-IbXq4gaTg81Z4GG_xNM6e4,29154
jla/distributed/expert_parallel.py,sha256=-sEA7ugoUWll2P4SXP5itwMTr64TBobqJ6JDhnN1T4s,12631
jla/distributed/init.py,sha256=h7OIy-4cw_hNrMrA8c5PTBZyTnqWX00MplLHqWHizxQ,13310
jla/distributed/kernel_sharding.py,sha256=9Hsru_Lj_9Xjq7NGrTqAvD7z_vNbud8odXKQTDa1HnM,24590
jla/distributed/mesh.py,sha256=cmtdbmqyIhDNLosElBjYdEJzW4zT5NUh16Aqwtc0FXQ,8320
jla/distributed/pipeline.py,sha256=ak-VEX3CrhNyeBeUu6JIlZcm57uEF6km_LbaBH3hX-E,28128
jla/distributed/sharding.py,sha256=rGOY5ZGya7RP2zSyzLrhKiRUBVD6dj93pCL63RNhWlc,11395
jla/eval/__init__.py,sha256=aS_Jfev-kjR7ABpeQKawlQZr_ZsUWihhF_1Xx8Byw-Q,527
jla/eval/cli.py,sha256=zkDaq-TGIVsMWb5FzWfp0VGPOJpk3dLJZZn2OexfKLg,6867
jla/eval/lm_eval_adapter.py,sha256=LDUK82Yvp-TbTuz6LHsznbhR4YitnJr8RVLQvkJTh3M,13325
jla/eval/perplexity.py,sha256=0C9u6equdFBOtdeYC7Lb2n7f8TUIYqSjk9w6aCmcyFw,5139
jla/eval/scoring.py,sha256=7d4kgJoWc4kXbvJ6IadQw-PKAZyUyFs7hCAY1MlQQIo,4240
jla/inference/__init__.py,sha256=WE0gwQmCQLfSjI9FvShZBVlRDuiVareXmBnxEBXZcIA,821
jla/inference/cli.py,sha256=k6_4n3DogKHmJmO8qZIclvZfn_t6fx9GLK5DxDpOgCI,6059
jla/inference/generate.py,sha256=HB0e9ugzwiDllhz1B97lwy9HAAYH_5gdsGGSUNPDeF8,13958
jla/inference/loading.py,sha256=g6W01La_hvHhCeEQUns9zrp17xTpEGAEhfYRiKnKzhA,9518
jla/inference/rwkv7.py,sha256=Sxx9YQZgvUuVeA_0MKn2JBpaPcHywbXij4ju4xAdvVs,3827
jla/inference/sampling.py,sha256=QD9H93LBPM_dcUEPPVYfZJMf6WNzaGFkm0cpkqZIc0k,2919
jla/layers/__init__.py,sha256=u296_xyus5kxbwn102t10rOrukaRDZtnn8uG9wGoq2s,360
jla/layers/attention.py,sha256=Uzyz_U8pfUJj4vg2oGL7diih08HeyapA65avEdomE2Y,26837
jla/layers/cache.py,sha256=s7YbiractF_UWUjshTRvQQpAThdfiku_jgKPPloJUVw,3701
jla/layers/conv.py,sha256=0YjZWRIPdjqQ-J2AdTI9RfdffJtaXjmSuqcYlj0G7rM,10111
jla/layers/conv_l2norm.py,sha256=w1mZtGHOwZWvHkJNhSLOKlrhVFLYFUFwjl7Q-NjXutA,5015
jla/layers/forgetting_attention.py,sha256=Qy91rcTB5dIDUHQDSR2ruPiq9R16aeGe5lJebCY-TkA,13696
jla/layers/fp8.py,sha256=-wxztReJSiSm3lpY1MQ8xEUoEjkfMMdsRagP3FbPT58,32079
jla/layers/fp8_attention.py,sha256=kh66lxSenztWZd6x9shPsf3jd33yW3OinGubU2X0Q08,20872
jla/layers/fp8_swiglu.py,sha256=VNLiVrfHXLoS3zwA409ZyJpswG_SLmEn329CBmS3-ks,23259
jla/layers/fused_mlp.py,sha256=fCWrJKFYuoSXxl4cBBTtmYvwdjVQo92oASPAwsUoDmI,4849
jla/layers/lean.py,sha256=nVONx-GHKIOsNTODvB9mqwbVcNKzBcd3Gcv1_EcHy3k,29203
jla/layers/lightning_attention.py,sha256=ZnTY5TGRPfb62Y5iKhKWdzRY55b7IpJGWsNR5bZTOnA,7946
jla/layers/linear.py,sha256=KSLUBtk-dFYbKbTi4V0IfRXMo311H9WSuFZCfkr-NKE,7128
jla/layers/linear_attention.py,sha256=ZnY_n_kwXvMFUcjvXCnGvuVRnJjtLG6ADCr1deSSsvU,32852
jla/layers/mamba1.py,sha256=s5Ne1Wl4ROLJndi9ANwpMEH_qnQpaCZI5JPwf3otDFg,7622
jla/layers/mamba2_fused.py,sha256=pao2FSkxRytT1g0M17Pe2YIAfi9NciTGf3XuM6toS7o,28513
jla/layers/mamba3_mimo.py,sha256=9NjsggOFgwF3KA9i7UW5ZbWhWJDSnKTVtv6-B_0UEK8,16781
jla/layers/mla.py,sha256=RlNDZ25aqvNfcYETk3h4T1RRPC-GciPQct84Vei9tIM,12735
jla/layers/mlp.py,sha256=9_6doquv8QmVPy8pcEzfnoS6tcd4T_dLsuL5oXu5Shs,2920
jla/layers/moe.py,sha256=nRUMQ59-0FbIP_zqo_0otq5G_JZdwy5XNaAf6G-h1s8,17592
jla/layers/norm.py,sha256=4zXB40GtNK_h35avBORnyuC-TX_N2BYKQwgJvWw0DtI,4293
jla/layers/path_attention.py,sha256=XwQwZetwy3if117wyP3_EkmTTAzJQ_hNxUsV8jTUOcw,5945
jla/layers/path_decode.py,sha256=BuWAumhjzGoR9h95afbzqib6N8-nQSZS-WerCwAAeEc,5873
jla/layers/rope.py,sha256=tngqZT-XlyNyMqTP--IYtPaDe4Oj9OkevbtBOQhjq30,6829
jla/layers/rwkv6.py,sha256=uwEIceJZIh-hNO72iwWgKc-bbnrOsmwOgvjiPm7EV4A,14904
jla/layers/rwkv7.py,sha256=I38gwQamkupYY8uWTiZGL1zJOSUHYGdHMn_3A3vpF80,30362
jla/layers/sparse_attention.py,sha256=Utfy6SstbqwvfWix016AK0J5EqE5x-TK6dGqH29ZAXE,11965
jla/layers/ssm.py,sha256=SNER9pmYifPmQf8E0PyVUc2XwUjqDSoIE6lsM6JaGik,36027
jla/layers/ttt.py,sha256=Jpq5F0A9PvYwzdXl35ngO5dWLeqbrBQsMhUSDAPqcqk,6732
jla/models/__init__.py,sha256=q1ha38iQnfyUp5L3cYRN0QYUNhf2jnkm_h0ffCsu4YU,1093
jla/models/axes.py,sha256=-oFeqDAdCSGZoW-lbSHZfZgsvdat1nanyiPDZQgQEFE,3821
jla/models/config.py,sha256=vNtGRGJ4DwDSvD_jlgTguU-RrPuyiX5sspFgGDUHnC8,49917
jla/models/lm.py,sha256=kSxT-HUJi1j9zoluyLkqveXMhkpQxKQwerRtTrGzjEM,34450
jla/models/mtp.py,sha256=d62uQklxOKSsQA5fwS2ahOlEisCTOB5PzLBj-bKZF88,8272
jla/models/presets.py,sha256=FnvbsKZwv7J1-5roSpd2eZd-D_U4f6BbnL0dxOmoDck,52056
jla/models/rwkv7.py,sha256=zqqy-cDwmtng65CoAncWss2IhScDd3gxC7bT3iz2j2M,7934
jla/ops/__init__.py,sha256=NJOUTJcFsAbxsFOjUAiI1HctM0xqcpJCNf9yz4CJ59c,7821
jla/ops/chunked.py,sha256=FGETjWlZtERbpqrwVwHWwaS20n7LiVQ7aBMzZHwvDGw,40699
jla/ops/cross_entropy.py,sha256=7CQ2XmM8UG8ONZWYBS9VmDmLAbbCp3UEUL64BdgFJqQ,22350
jla/ops/dsa_gather.py,sha256=TbaKQpacQnlU3mgCHcwob_JFIvep8qYffBnwWkylTnI,4485
jla/ops/log_linear.py,sha256=walNH61lTE0mCqOYnBVvcNQyzHQFfX3OC2-WSigu5hU,11677
jla/ops/moba.py,sha256=2gmUjGnKWyNZrFvJQrsaHQV9m9MOmPArsXN0SliuqG0,8746
jla/ops/moe_permute.py,sha256=x1nI0CRvpPax3J3lk2CiyosK2sLYDfqq-NFGEyR8QKU,5547
jla/ops/path_attention.py,sha256=BVB9ovu0Gx-tleBeCwf2F2PUbVa7Wz2CQ3fwHnNilwk,10442
jla/ops/reference.py,sha256=qjtKVG1IRtNtEOTwyJ4N2guBP7B8xKc87GcPM-2hx0E,33963
jla/ops/rwkv6.py,sha256=A0v3fiP3AtWsP6IYiUiQyRGAZQLfyLBpzSaORAbhBzk,7083
jla/ops/rwkv7.py,sha256=ik2gCTc0ZJCbcHpl4GC5QC-vFfoOqxl3KvKtJGdysQ4,10478
jla/ops/sparse_attention.py,sha256=3bUakl_Kf5hz5YE5j-1D5vtAVTtg7Uz3dY2ypkAyzPU,16085
jla/ops/token_logprobs.py,sha256=1ZyPcwghicC6dmoFClaruaiaerQxRa4OUg1JppBdbbE,6468
jla/ops/ttt.py,sha256=IxmkFUMhxQuPl2Auj8gctLcPzJUyF8kN1Ag5Pa_9B_I,13933
jla/ops/mosaic_gpu/__init__.py,sha256=ucLIWn-XbIb9NhuUwCQmbeUFF0-5-h_M_MeVW5HPRD0,1674
jla/ops/mosaic_gpu/_common.py,sha256=c5FGkOw9yu1Bjbp_8NnspDNss6K0nqj7IUYKnt6XLko,8473
jla/ops/mosaic_gpu/attention.py,sha256=8r-HK_uslPXlM4wW25trXvFuxwh3KZxxziJjTcroAKc,105628
jla/ops/mosaic_gpu/attention_fp8_bwd.py,sha256=u3KPNFxPYWdOXZCv1EJu2jIn0NqMF1PNCs-hfMbGRZY,46969
jla/ops/mosaic_gpu/attention_persistent.py,sha256=gWDNZc2x_Wxdy1f0pglOxzd_NAk0XkIIPjbqfBBOWkQ,37098
jla/ops/mosaic_gpu/dsa.py,sha256=XQMR35nh4ZDEEfXTicRGk3LeekOyEg_Nt_HeDdgT01w,77926
jla/ops/mosaic_gpu/dsa_topk.py,sha256=YXaRvg9v8GyFSKcmRiSlTd4Gn4PuldmoMWWC6jpTt1o,22884
jla/ops/mosaic_gpu/fused_gemm.py,sha256=Utcg_5GIm-vTgP_X0o4RQfwm00BLwf7jUwF2vkGBcCY,24032
jla/ops/mosaic_gpu/gated_delta_product.py,sha256=TmFXrxd4kdO8JJ5nRWclemvg4SEcbiDE5MvhDEU-iZ4,3578
jla/ops/mosaic_gpu/gated_delta_rule.py,sha256=IDYu1VPs6aGUTRFDmAN_CJTDzluwkzx7U7Go56_fQvA,4404
jla/ops/mosaic_gpu/gated_delta_rule_bwd.py,sha256=0jNvV_F7TPgbSPvXexsSimUjr7ZuWkqjXuSSLyH6Ulk,40480
jla/ops/mosaic_gpu/gated_delta_rule_fused_bwd.py,sha256=YMLoXh04eKZDxSOZFzzKgX7pBRCX2wh7E32PWZ1kg-E,22030
jla/ops/mosaic_gpu/gated_delta_rule_kb.py,sha256=uiT1EXwob-wVTaM3znZHd6kXRgbnq0ZOxrOZ7LaLBMw,55602
jla/ops/mosaic_gpu/gated_delta_rule_kernels.py,sha256=CBDN5q7eFifcHDN1QkR8sOZmcqpxWKxs2ivaX2jmC5Q,30413
jla/ops/mosaic_gpu/gdn2.py,sha256=524ekoib8HqwzjdZZEtEWZsOpaTa3kxa4dBNVQVefuE,2388
jla/ops/mosaic_gpu/gla.py,sha256=sB88pVSwT3a6ySJRdCjQjs3BQQ0StAAz2CtITacY3Xg,6830
jla/ops/mosaic_gpu/gla_kernels.py,sha256=D64ewbNTHxxcaNvGa2ykxHsW4Y2U_VAqE-M1aW5iAbI,44876
jla/ops/mosaic_gpu/kda.py,sha256=C3rqH62bWA8WTpVOpSk8HDJCVgQCHMgOgRIbPMNK0sE,2057
jla/ops/mosaic_gpu/linear_cross_entropy.py,sha256=RHCzLrpkTpz0XCG4ZdmXJXQ3DJf9zNb_jlm1L1WhEXY,30609
jla/ops/mosaic_gpu/mamba1.py,sha256=185Q2lJgvEkHp5SxTTBgOyAC9CHohcGcjLQvlyTSQvE,23666
jla/ops/mosaic_gpu/mamba1_bwd.py,sha256=BJ5vCdxNjjx59Mz3Sw-gavkpn4n32cE4_4_hKNoEC7w,11642
jla/ops/mosaic_gpu/mamba1_bwd2.py,sha256=dkhHfjQsFaj7-BjgU_j6S2F-Hw9GLqqs1JyAvClarqU,25785
jla/ops/mosaic_gpu/mamba3.py,sha256=bAtoUhGw5cxoSkyJRoQ-kWRBZvtCcuSwkNpqEfzNy9Y,43583
jla/ops/mosaic_gpu/mamba3_fused.py,sha256=CoZdU2JtLqcLcA2jC2Iv1pYHsyfaZHP_O0IJitUj_II,36938
jla/ops/mosaic_gpu/mamba3_mimo.py,sha256=cMN6WLSbLlNqv8KqkHigzpjYkBvEq9KMNajvM-u7Roo,8081
jla/ops/mosaic_gpu/mamba3_mimo_kernels.py,sha256=3W1OGGDMy_5esd51ZoT9fEVJ9flob7zv_XcvCrxDv9E,37128
jla/ops/mosaic_gpu/mlstm.py,sha256=ftGjyyYSRBSPAjIuGSYNGOCPhFnv4VtMehJJOak0XrU,14725
jla/ops/mosaic_gpu/ragged_dot.py,sha256=faC9IvlhL9xUG6EmR0aDNv5QSJaetAxLfTYjVDj7dGg,44952
jla/ops/mosaic_gpu/ragged_dot_fp8.py,sha256=LSHmHH-gaNdZDs50_8T8HbFtVK_XmkVnix2Rs4J14lk,6019
jla/ops/mosaic_gpu/rwkv7.py,sha256=-83ObpIigcbJRQncfldnZLuRbt9BvIEJMD2wE7O12bo,41802
jla/ops/mosaic_gpu/rwkv7_bwd.py,sha256=aI36HBDFNUfqpW0QNuSQ3MM_8pKadwpExQxpfQ1SW68,21447
jla/ops/mosaic_gpu/rwkv7_prep.py,sha256=1JlN1tpKwyUkdk0FuGVLnSvm-ZgrkAtmlcl5iL0UnVQ,20306
jla/ops/mosaic_gpu/rwkv7_tiles.py,sha256=pC79EpPHCi4Lzs-i7pwe26e4M8xrF_WHExEXM9yfTCU,4157
jla/ops/mosaic_gpu/scalar_gate.py,sha256=vZJwtHfpbmv79TKwL6-_ieIt21BCVJhH5koKcuCILxU,34812
jla/ops/mosaic_gpu/scalar_gate_docs.py,sha256=aAS2m7LSsQINzMr9NEFf1HcWI9LcM33MC4Olu2mDT7I,45021
jla/ops/mosaic_gpu/scalar_gate_fused.py,sha256=GRN8GVYG14pFDdWOiShIN5RYqKB3K7CDVqiM1SYQVtw,23730
jla/ops/mosaic_gpu/scalar_gate_grouped.py,sha256=v5FkVxV3taBdeK5K0_fsCWsiwnKNpRmf-ICETsdUQhE,10947
jla/ops/mosaic_gpu/scalar_gate_kb.py,sha256=oThomKpWfAYA1x0S_IVuEvgzYTohY-LpCARJOZ27Pko,19220
jla/ops/mosaic_gpu/scalar_gate_kernels.py,sha256=pvgrMCs-AYUzKK1Pkn7HxTF077M-q2JAKGc-KkM7z_4,26234
jla/ops/mosaic_gpu/scalar_gate_scan.py,sha256=lRvgSuKBk974zzFnrDD2fEcKkquj3tkxUEPGxZyQ7iU,8365
jla/ops/mosaic_gpu/short_conv.py,sha256=TpV-tNCOPcvZR_66T3wsgnooOauOG2or084ZO4vItP4,26013
jla/ops/mosaic_gpu/short_conv_l2norm.py,sha256=DQ5Dks60D545CEOY4phLrkiseqC5vbIkuOXF9Ba0AUU,19652
jla/ops/mosaic_gpu/vector_gate.py,sha256=a3j2QsuCROibB3qda0gZrzgt-zle1rrTQFqU_HP09JU,61821
jla/ops/mosaic_gpu/vector_gate_bwd.py,sha256=GBPvEGSGs6Zg5VEC-3D_zLs1Fe8hhvgPPZ-e_FJ_sSw,38878
jla/ops/mosaic_gpu/vector_gate_ops.py,sha256=ZCvsilEOok1HoGlVBEJggdFU6GeSx6s3eG25bJYMElY,18039
jla/training/__init__.py,sha256=VExQx4-QIzbbpE24kh4p_vu_uhVx8VAyK8i5s7uwch8,364
jla/training/checkpointing.py,sha256=3hAdAEQDiQeY0JDk0KueLMB2Mq_JDDIRcItVgVR5gJ0,31108
jla/training/dpo.py,sha256=xc9OC1KmDA3Y13NEn3yE0Sjz0cjE059ZhEOe0roU0zo,40117
jla/training/fp8.py,sha256=NJieTKe5ziaBBE1wUBsMOgSU3RE8ntpCrEHqRInyUrM,7996
jla/training/gpu_setup.py,sha256=TFthJzAjxldpK8fBx7RWxAI9c8_OXxzZz6WbMZ9FhXE,17353
jla/training/grad_reduce.py,sha256=TbOSTnrcdjko61ZaaZ9gDzjo3LukZPnugOmWb4he4OA,14855
jla/training/graph_health.py,sha256=9V5Xi7VFSNH1LBCyPj1NusEuBKFbkY1fUyjSEjoCqqQ,35726
jla/training/loggers.py,sha256=HkZPYT1MgKeZVHdfCIKHBwLuep9n8Otr8aoouZ6GVu4,6325
jla/training/metrics_sink.py,sha256=qGcPeTwNERTrvu-8xRcs06Lw3OetmteqwKLYqgrO8pA,6256
jla/training/mixed_precision.py,sha256=OUXlv5TgvP7NdxDjDp_Mxnzm0Xxf7_Cor-N39BqGW6Q,4813
jla/training/optimizers.py,sha256=1rII8uS-b9RzsLIThmsY6iHQj4pyxuLleVr9-pw7eVk,25765
jla/training/performance.py,sha256=nfrBLbGzL4Cqr__oGaY08JRjVoMtzmw8nKH2AlGxdSA,31009
jla/training/runtime.py,sha256=unIUb36pqy-5ntMUVdR4JuPxL4yFmCH8fTXmNF2YQdI,16271
jla/training/schedulers.py,sha256=Zqoh_WYhV6DD5amsp5HK6SvhsZMdNNEXWta9ksE6OCE,21681
jla/training/step.py,sha256=2VkJ80KAgJT4CiEu1RNfGaAEcXsF1RVPyIG7pQ0FlYs,27016
jla/training/telemetry.py,sha256=Eb5542TFEitdpt4Dy9BpwpCrmnuVI3yivpcsfPWu0UM,8467
jla/training/trainer.py,sha256=MCptg8gtFsN1CyNgCQVJqTQ_CK1D3hTNMxqO_pmRhQ4,36472
jax_linear_attention-0.1.0.dist-info/METADATA,sha256=Qh-_bcdPiWRv-EW4qbMC_Tyqov-T4MzHP9kiVaP1zuo,36248
jax_linear_attention-0.1.0.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
jax_linear_attention-0.1.0.dist-info/entry_points.txt,sha256=7QiLsE10-1fAW65flmjqj5PwQu433IMnobbB2fc2gQs,87
jax_linear_attention-0.1.0.dist-info/top_level.txt,sha256=gVvJyaEFOQbTc43b2G5PoOMGWHbsJ_kY5W3db6rkX5s,4
jax_linear_attention-0.1.0.dist-info/RECORD,,
