priml/__init__.py,sha256=CX8L269JaOX882Ygzn8kbw-QaXqVlKRsrhdVDY0buDA,71
priml/compile.py,sha256=QIMI6cDpdtZ0bf3U0VdVLJES06TYp_Jxxcqq2Qw3itE,4747
priml/custom_types.py,sha256=Mk9nkhWfoivISil-mMNtT5w8_dXmX7rXuktKcVHCgi4,2595
priml/hub.py,sha256=AkoqZ5Ih72xpZxgCo2YRP8pPH4r10sqAZUJUHHSyvN4,6676
priml/image.py,sha256=PqDCDmzif8SGh6WUehItI-2idnjxVawnWmrXGsjWCR0,12790
priml/launch.py,sha256=NuT0c5lZR-zn4kfNI1MbO46QdD84mJSTmBUFYTnpL7Q,13444
priml/logger.py,sha256=fvEwxa_-holK7V66x1RXQMxFHRnR6gKfwpTiomZoQUk,7537
priml/runtime.py,sha256=tyii41B7d0tOr__zBd3WPtIPFtyTx1-YmxYD19jYp-A,17930
priml/sharding.py,sha256=GICFFTBtrBtsc4FatxQyzGzHYgATd2bIIa68ohJ6dxI,2417
priml/data/__init__.py,sha256=5nY6m6LIZKbxl_wi4E7YF-W6pvFKU3B3xkLH1bNcoRM,71
priml/data/custom_types.py,sha256=bhuAaRm6GM1NpQJsk2iWMxdCMrAgkx1h_3esbsadx9c,881
priml/data/dummy.py,sha256=NPbxCnwX3CTxE9eNegxz-GW2AO16eXzSVwhOE3mwlnk,2921
priml/distributed/README.md,sha256=hVjVh8FBFImTdSeoYrsOYnXUkIyxfVfolDEsGhKFTbQ,6040
priml/distributed/testing.py,sha256=5X5GxtNCjop96HfoHVC_6-1iFTEUYtPaAeMu5nGxql4,16369
priml/inference/__init__.py,sha256=iNAc0MRRcHbzfTYdG12-DEmPRGzVm80AcV6esFM4x5o,61
priml/lib/__init__.py,sha256=8gOpaeNpBc2xiI68JpniMb4xyiviMeh1ueOEvYAfakc,82
priml/lib/custom_json.py,sha256=YKNnkoAG5aKz3o3ovNwtyRxAEFaECqvoLU_1KmFqz8g,27748
priml/lib/custom_types.py,sha256=Pu0iXtMSBT47sax2pRXBjfXaFRwX4PGJPCfKUUzliG0,1499
priml/lib/traverse.py,sha256=jogFjY5C0wv0-RgO3cWobaNZ3bKHMq9P6OG0SAKyP6U,7174
priml/lib/userdirs.py,sha256=6HVHxMM9A7AxhU4EDHqde-S0MZXkQCIDAQ-OnS77DKs,6500
priml/lib/testing/__init__.py,sha256=EGIErww_48bLkGtYLrTBy4hnXYPuCdXZ63upKFL1zMA,61
priml/lib/testing/main.py,sha256=sKBvJZC1H1ROvBVb7gLdjeaA68vM-1in6mmmEPyYdHQ,1046
priml/loss/__init__.py,sha256=K2BKG-PBwiug96Nja96ZG6k2QbyHmRKq8MRf8Z8M1YA,611
priml/loss/cross_entropy.py,sha256=qVMsPvhW1pvNxMMitKoN13srHJUzvSYme3fqToIlyYU,2552
priml/loss/custom_types.py,sha256=K-y_65TL2DH9Rhcbsy1DGFyOLm2F9HNDvxlLBWO2Lq8,1262
priml/loss/diffusion.py,sha256=OusohDzWkr52CS_5izXTdO1LTWdAERK6lCu9lNTmElE,5094
priml/loss/gan.py,sha256=5wGsjtxjXgghaKq3FjWzmu287CdB3jHhMkS7h6gyutQ,2292
priml/loss/lpips_loss.py,sha256=ZkRVYCxQ1XXo3V0cu-1IlKzb18y_mZcbtqb1q4Gn_BU,2358
priml/loss/simple_loss.py,sha256=HVEr52DaJdIbCU--0htc28VOtY8JzFtcANO1foDjcOc,2319
priml/loss/stablemax.py,sha256=7RsvBKBXs5XHKpafLYEtpSTniy5guT2vIhh7hjbBbVU,2118
priml/loss/weighted_loss.py,sha256=ovWXkZXr6oLjNIcTYTkF5RYVXmMQGkVxdoMv4YBiJ3c,3271
priml/math/__init__.py,sha256=6AVqEnNppp6iRlFa97eTDgRcbk6x5NywgEHaWDsFAzc,58
priml/math/basic.py,sha256=2VoqR2BDyDMT25OOsUgQyF5RE8XGwgKx-_7bLY1qgwg,5746
priml/math/custom_types.py,sha256=U4cJYEEKkxhtVl32-G0f4eaZz2vcSWj4jRL2DdsTT4I,6558
priml/math/distributed.py,sha256=O9sQf50WnNYEf9PHuYKXCtgNAkLJvfA1t2AFwiv0EvE,2443
priml/math/frequency.py,sha256=ACFL4sY_J5DKcIuheSiBX6iwU40kfBY8a8N0QqX-iQM,4341
priml/math/numeric.py,sha256=NzjHdHybyQhXPK2KAcgVLs-mHmAacWHuN4ljMWvbP0U,26664
priml/math/pixel.py,sha256=NyEYtLMCVyM0vx1ZODemCEgBjv5S0H751M99HtQPoXA,15446
priml/math/pooling.py,sha256=AHlHMv7_74FxnJGySOx-HOuzAf64jzm6YB_1jrojfno,7232
priml/math/probability.py,sha256=WdjUKpZouKd1V_HSofqtFSfd6YZxOeEfidMmS7q0kXY,19724
priml/math/seed.py,sha256=6tUAA80sj6VIqG9jpqF7UJdt-ws3MfHDkf5_8_XkqqQ,14424
priml/math/stats.py,sha256=I_u6eCW09QW0rhctrtoM8eFc9_vZ9CbFc79eob6reiA,13640
priml/math/tensor.py,sha256=FsFF6b7WKMiqj3NU9WHgvu3pnJFN9ZZ55GEFx_IcvbA,5818
priml/math/diffusion/__init__.py,sha256=nrGMPE9zirx-8YDBn3pfeGhXrkmxexjn0MxWPIMKCss,2032
priml/math/diffusion/sampling.py,sha256=HBIxmfQpIvZ7a01whq3mlMEmrcBtaiC19KcCce_FLDE,15036
priml/math/diffusion/schedule.py,sha256=evKDbLlMthaTnh80jFmM-LybOxjhtscOoz-VvK-QbyI,13916
priml/math/diffusion/target.py,sha256=jGdSmlbJpAgwtvCZ-Ou6WKOFe1aI_jzr_7zUeOhiVk0,12002
priml/metrics/__init__.py,sha256=JUz3QpWm7e-2g3AQq2HZAk80QoKXSX4enMDOrWFF1sY,271
priml/metrics/binary_accuracy.py,sha256=JkD_YsSqjcbZkUEkN5LO12wg8sR3SVK-TopYg1TfTIY,2193
priml/metrics/custom_types.py,sha256=DDDcsePXQsxFtfOdBw0rOg6RygHLShiLWB-jcop8goQ,1226
priml/metrics/topk.py,sha256=yl4TtQ0jc1IYHAPJMX2dajppFn0VH6JJhSJ595QE1Qw,2809
priml/model/__init__.py,sha256=buv5CFeblPLn6te0SxPavqHf8X_q53Ek8wp-QXCcqWw,2142
priml/model/attention.py,sha256=7-mL3UQxVWtUNQKFJZV2c8ZYmIYqVNQ8f-PBtl3bMg4,28149
priml/model/causal_lm.py,sha256=RHj1qRDQheexj_joxwu_bEZEIknCsr-KjJx5B6kLWhk,6773
priml/model/conv.py,sha256=3-WNdcxtxlOyeFbkRnexYzAe8dDSNgn1BhmKuDAJRIE,8067
priml/model/custom_types.py,sha256=AbMcftNKGFZdBI1-5fipmdtgj09gNWZxXCoqqxC6n9k,2750
priml/model/embedding.py,sha256=xKS1D1fk1fxBsFdhqbuvrt8mLHoFxHMrLMiUBsqhiB8,1528
priml/model/gated_delta_net.py,sha256=7jkQcTHBSAW1BiIEQ8ZELWN1Tp9M8o5iq9VeArPW-UE,9071
priml/model/generate.py,sha256=eT-e7auZSkRE-Wb4cQem5uGikCmSjK1Qqd2tuVeHAzg,5702
priml/model/init.py,sha256=1jQYjoUkDXoqqPy53weJ9FN63zi63KiCmw9_lRyXbhM,2462
priml/model/kimi_k2.py,sha256=7gMlOqLJ3UVDhp6uaM_DPmm2Pwu2u6ELN4jbZAUmPi8,15947
priml/model/kvcache.py,sha256=-Ihf7rSpsi2oWPOjVKDqkMFCtNHgF6mPvcya3hN9f1U,3817
priml/model/linear.py,sha256=ufew_yWOmtCna_Zk_GFW2MnQ3sYHCbU31QMKkubpI5E,6700
priml/model/mla.py,sha256=neBW2utp0FLrbuXVeDQQa36stOfc_YW8MkoiGmMGrY4,20391
priml/model/mlpmixer.py,sha256=4WPCDxTWS6SRwTQZ6Gldcwq8IYPg401XnGoY74JGlFg,3722
priml/model/mmdit.py,sha256=Y_fHjBHAef8lzzF7V8rc-lD_frdcsyeL7qbwK4UVabM,7720
priml/model/moe.py,sha256=PwmWyCpiKWYpiC60ppu7HO3OpEqa7SRmEKYjkpdPiRo,15520
priml/model/norm.py,sha256=3Te5mworpf3d0oVdXJFbrMzPcuQMlXxk96GdfMKszmE,8978
priml/model/patchify.py,sha256=i0E5P58W7iNxwNmTJbWRNhVWgYV1syxp-jmU5IM3shs,3923
priml/model/qwen3.py,sha256=IJXXFuAAT0cmGk3uMTJ7u8aA3xABn8apl-tJkqDuBoI,10370
priml/model/rope.py,sha256=By-DyXQl4-yhMT2nNmkndeo5ml4uu5MRVPeAX_o_jXw,23576
priml/model/sequential.py,sha256=NniD7u9ottfw_AT1_6u4MbGu4UyuCCfW908uvnJdGOc,4243
priml/model/special.py,sha256=eRrXm-0PGWSpDbbbEGjewFQNp5sFBBlqAgy8-39Ffow,1772
priml/model/swiglu.py,sha256=MS4MnNfxqm6sh74LQ-rZevrfunUCciRWmQM05MDRHuc,7613
priml/model/transformer.py,sha256=OGxdnRrGQO6J6YI_cODJ5iGcXRqOeT9tys9K-KM1Mow,4950
priml/model/whitening.py,sha256=n1wpow_Jau_qxCcylj6QqlhdEy3pmyGLkh_cxgcKehU,1938
priml/model/goldens/gated_delta_net_min_cpu.pt,sha256=s_rM4phhAlNR1f9WIcFpKpcwolcoGBnLb4Y3mhZuWFA,20423
priml/model/goldens/mla_min_cpu.pt,sha256=ysm0AlPckxR_7QnXZq7t0V9X2AMzk8-5NaxTxXigp2g,14407
priml/model/goldens/mmdit_block_min_cpu.pt,sha256=2I1RADznSAF0_RlnFrISUHWV5TYxqzlzhXULw8OC0fg,221718
priml/model/goldens/moe_min_cpu.pt,sha256=keFG-UDbNVBH5UPloS9Ol0jA-eZxf-Ib0IF7tVi6hKE,401807
priml/model/goldens/transformer_block_min_cpu.pt,sha256=SVpyihfTCuAatKlUlDDxynt_moQSjPn3rDGQN6HRSZY,112037
priml/optimizers/__init__.py,sha256=FZm850p9eh8zEZ7lm-pwPUWOheKCC7RWz-AIdbpVsYU,3594
priml/optimizers/adam_atan2.py,sha256=MOa4zdmLBabvDm5beObx917DqCzeLGG27aqqCkYRI3w,3864
priml/optimizers/muon.py,sha256=peVl_1xcwPMSp_ru54dTufIsenDPhe9sVhwiwhGqvjo,4970
priml/optimizers/newton.py,sha256=aJSHcA6AVwKCOF3HbtA5dkMobFhh-WaPNhTmbMxykCc,7523
priml/optimizers/sign_sgd.py,sha256=S5nMPG5k-xUJBIc9zj_k3ttxNE4y19xGvF4St0Y6Sno,11357
priml/testing/__init__.py,sha256=EuGRevcV68b4POes0wZugnbhQ_TXUR_mI-XJUQXCq5I,44
priml/testing/bfb.py,sha256=4F6wsdebo6RNqgM79ktgwhM1KmjoFUFzTdDkjQ_HM3A,31173
priml/testing/fixtures.py,sha256=W6uCZZjiVnc80A2_Ml2xpDBDDEQ-IQ-pJUer4S3uY_U,1243
priml/train/__init__.py,sha256=XM6L3vVPA1X7q8CiBR2oG66NYsLAfjL04G0WVIqax8U,1849
priml/train/activation.py,sha256=nG3kzB6nK_LxKwNGDBdKz8o6dTNJht7TCVzUSx9Zbh0,17791
priml/train/checkpointing.py,sha256=ezykX4C-p57irKhiRidJYoBynP_2Wpmi1NRJ8f5sZ8w,33439
priml/train/custom_types.py,sha256=Mhw4wrR49nd1av_acXZgZbLJXxYDMns0cFSumCCe1lU,14813
priml/train/ema.py,sha256=jXKLnb42S3owRvm2kgctCbicr6RAP3FFJ0lpkvrhkuE,20865
priml/train/grad_clip.py,sha256=EHk73eIA71vJrCbr0i4_A0A6x_eOg5E8KpLJLDqG888,4690
priml/train/learnable.py,sha256=cLO9JCpx1kbwDGQLuMNGXj0_EKsmt96YuJjLjhs8J2c,10764
priml/train/parallelism.py,sha256=BKzGKVzPydmRP2s3thknn_Nb7DsloaXuyAvucy3Mfas,19062
priml/train/profiling.py,sha256=8WNGUYWFtJqwgWvATOK4W7eiFemng2r0fmO7LLty3DA,20219
priml/train/progress.py,sha256=dz7FOB-MlggrgJJIKo-66ykcuV1G4m5ePvgd5l2bOpk,1925
priml/train/quantization.py,sha256=h-mllP_xbVakUN80s8mPvC5_Ed1GGq87PJK2BCk-J0E,5890
priml/train/results.py,sha256=u4kfiKOLx8iHHgOaNA7JIlVqxBjRkCWUHjhVtp3DILc,2895
priml/train/schedules.py,sha256=__ubJqLO1z_GQzYFv6279dxdWAK5It0v0vYbUzcU0Eo,2716
priml/train/tensor_parallel.py,sha256=MMJzPAz0QhA1vBDae3NHaQhwn8VujteyX2nzIJ_hU1c,6966
priml/train/tracker.py,sha256=lFSinZxvGMJ9u_7szDDEtbFdVXIVj7d7L4tUhB-B-1g,21410
priml/train/train_loop.py,sha256=lQI6AqzD3W2n9Z9k7pS_YHM30OZ97DkE2a9wXizYx9E,59867
priml/train/train_step.py,sha256=h8DhKo2tbs7WVKG7I0TEaUNNGIuQ9fcbldE6i51Cxdc,13376
priml/train/train_step_gan.py,sha256=x3jcZOtM8Ov1xbEwFSq_KSXH4ppau2v4Y707Dnhp1GM,11607
priml/train/docs/mesh_native_design.md,sha256=W2SySzJQB0KHDj3vb9HhYhyYAkiSq_4sTUPWMwdCPDs,10564
priml-0.1.0.dist-info/METADATA,sha256=WxmUy9WgZEwTdV-_ivNrmmMJC2pZyiNtGD4-FBifuuM,2561
priml-0.1.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
priml-0.1.0.dist-info/licenses/LICENSE,sha256=UOZ1F5fFDe3XXvG4oNnkL1-Ecun7zpHzRxjp-XsMeAo,11324
priml-0.1.0.dist-info/RECORD,,
