priml/__init__.py,sha256=CX8L269JaOX882Ygzn8kbw-QaXqVlKRsrhdVDY0buDA,71
priml/__main__.py,sha256=4wa8yosSUNZzMkXebhO9K067OF3RrlxJWhx8S3_7MoE,410
priml/compile.py,sha256=QIMI6cDpdtZ0bf3U0VdVLJES06TYp_Jxxcqq2Qw3itE,4747
priml/custom_types.py,sha256=Mk9nkhWfoivISil-mMNtT5w8_dXmX7rXuktKcVHCgi4,2595
priml/hub.py,sha256=Ajg5pi80awJ4yU6fiB3EJ3s2GjP_3NoRDlURn5ShumQ,6836
priml/image.py,sha256=PqDCDmzif8SGh6WUehItI-2idnjxVawnWmrXGsjWCR0,12790
priml/launch.py,sha256=0xZtpy5F0ao7enOKcF4EuWR29tN_Z-PYDu3hOJ6LfF0,9533
priml/logger.py,sha256=fvEwxa_-holK7V66x1RXQMxFHRnR6gKfwpTiomZoQUk,7537
priml/paths.py,sha256=k2xnyAl_xudBg9Uac4-YsQf-MOCY7qs9-FgHC0l9naM,4970
priml/runtime.py,sha256=Q2bpy8atowVHOqGIyJSyw6wiyU8oQI8xQzZta6jy0q4,18270
priml/sharding.py,sha256=GICFFTBtrBtsc4FatxQyzGzHYgATd2bIIa68ohJ6dxI,2417
priml/timer.py,sha256=AgpCNf9T1UylbnBAaNYp2MTNJOlZ1nmlVdUW9tkN9Pc,3884
priml/baselines/__init__.py,sha256=fSfKpHn7aQkbOvdfSw0uKzJtvY9X_mAGYHFdVo1Ko-8,518
priml/baselines/arcagi1/__init__.py,sha256=OnkVYUup0G3eoyKGH_FY-lc4bLGZ3J4c2_S4oQFy0AQ,523
priml/baselines/arcagi1/data.py,sha256=rDiCGdGW7rzubhNOiILyfzeYSQ-toxwGyeWI_D_vf9o,13839
priml/baselines/arcagi1/experiments.py,sha256=JX_tOAevb987PlxCHER3KKQX95-DEsyTrRB_s-Z4dUg,9063
priml/baselines/arcagi1/metric.py,sha256=5wa4-UqgtJmmiB747rlgBaskY81bFHXD2WeSpUdyD1E,6159
priml/baselines/cifar10/README.md,sha256=ivqxCUlyrzj0Q-MoJgDenwZVJk6LVCxCvI15KvNtn-Q,3956
priml/baselines/cifar10/__init__.py,sha256=v4AVFr7dPFyvBV75zIY9A14D_mdHAAvYM03P2nKOGHo,265
priml/baselines/cifar10/data.py,sha256=QMVHvGuakNEPHQQuRIutbsTITbBpRpbwFOKa5ZGsueU,9788
priml/baselines/cifar10/experiments.py,sha256=nHr9ANgEeAG0jp031VDdX-DrSLMxLaki6u-tBG86yrw,9246
priml/baselines/cifar10/model.py,sha256=REWz_IeHNTVogSclGfrjX31kUZ5rpd7JxW3_dNildy4,19753
priml/baselines/cifar10/train_step.py,sha256=YOT2Hv8b9zbgTIG6G8v2C43QjErqAYwTWRIFLS469Rs,12705
priml/baselines/cifar10/goldens/adamw_three_steps_min_cpu.pt,sha256=rsF7ZUHF_w99iOaBaZ8Flm7gxNrATIm7DLnIJE8PjEY,67738
priml/baselines/cifar10/goldens/muon_three_steps_min_cpu.pt,sha256=SIIAjzZCXG5biKZS48Lx8CCzHVdDs7ZVl7ib5TzcszM,67663
priml/baselines/cifar10/goldens/resnet_min_cpu.pt,sha256=REY28BLxPNrSqCA-8LEoOr7lMdbi052QqTQx1C4J9-g,64605
priml/baselines/cifar10/goldens/speednet_min_cpu.pt,sha256=tyPAHZim_NP-MgeMURNmOBzUXgD4OhWjjkULWjLhr_M,77709
priml/baselines/cifar10/scripts/__init__.py,sha256=eve-p2busGZK0j-IalrVAzLaDnjcaQObDjDLC2Gu4hg,52
priml/baselines/cifar10/scripts/prepare_data.py,sha256=iwre4n7_fvQbeZdyeMPpbK-CZx3S527q_upFMXjAUqA,2027
priml/baselines/craftax/README.md,sha256=qE_HAgmjx_DiSane8x2pTPnmkROGat6r6PYkUzpOJ0s,6657
priml/baselines/craftax/__init__.py,sha256=caWZc7JvbY9Y7xFyv0nEyEe-R5TvYrrJ5zuni-XsaxY,548
priml/baselines/craftax/data.py,sha256=Ds4ndZvnppM2236aVzbmND-hBP-ZS4mCCK7igubhG64,3839
priml/baselines/craftax/env.py,sha256=4fRTmd0tAoxrTAtIRMElR5-7YRIN1KS4KzD3UQAIBaQ,9755
priml/baselines/craftax/experiments.py,sha256=UWq2vRsgJzZL3-qyxwYwuZYTCwNO7H3yBzBpZi26HXQ,18043
priml/baselines/craftax/gtrxl.py,sha256=5ujeATN3A7VLZHSuNVDFhrsSKN18-XnCMfaZC901v-4,21386
priml/baselines/craftax/gtrxl_train_step.py,sha256=yslBYcp65DkHMzYDC_eC7dhLyVdPlYOBDXUX-3yhlCM,27261
priml/baselines/craftax/metric.py,sha256=a5YIR03QIU4klcaHQOPCM3Lr7EBUNB5TB_fwJciaV4k,7232
priml/baselines/craftax/model.py,sha256=r7FxVhIKv5KbeNXV5eleYJsMZc66-8fC6_HIO7Q8a5o,3921
priml/baselines/craftax/pqn.py,sha256=-XPyd3e9cL7DWCacff8H8ImazY34_STqG2Hwa0shnrk,8582
priml/baselines/craftax/pqn_train_step.py,sha256=KxrSZX-lRHtR8Bo6XW9c5ePz7t46TuYm6BpqdgofYPo,24377
priml/baselines/craftax/rnn.py,sha256=OC_PD-JhXhtKVjNXWAIyq1cjKNiEjuGEZ5JzwCaJUo8,7478
priml/baselines/craftax/rnn_train_step.py,sha256=RG-AcTNlq6XtsOn3k7Dy2AovvO3Llu39w7eLUmqcXsc,22885
priml/baselines/craftax/train_step.py,sha256=JkPjTTXfXPrpMAstikpi_x6Fb3JS88UYyfxBshb8vUY,19261
priml/baselines/craftax/game/__init__.py,sha256=IxJeF4Xbvx4uiJn28iP7UlKFd4xbWLiqoxRGF9pwyeY,803
priml/baselines/craftax/game/abilities.py,sha256=0SMbFYFEJsvzLXdbxcXfpL4qLW6W3XSGrQD9bqvomD8,12447
priml/baselines/craftax/game/constants.py,sha256=uAolBV-G-60Qndj-_Y1OBXh2m_sqOSw13i3IWxNSzR0,17617
priml/baselines/craftax/game/crafting.py,sha256=5eiD54Qg4HQWB-Isc2ty7g9VnI-2MIW4yAJ1SGzB_Jo,14141
priml/baselines/craftax/game/indexing.py,sha256=2lh5v7bfTd_Lcx22AptddImn5udtse-5T3hR2evK-Wo,6873
priml/baselines/craftax/game/interact.py,sha256=iK3VudojJuPMgO37j8LKCpN207Tt3IZSFyU5yx-lJOY,11260
priml/baselines/craftax/game/mechanics.py,sha256=e8ob2CONbuNNjMfN2qW8eObQu0yxOh_PgOXYmYCW4rQ,12094
priml/baselines/craftax/game/mobs.py,sha256=BmElTPmnk3uTOf4Rokm6iDmW2tkZADbd7KK5yY4mRLw,19234
priml/baselines/craftax/game/noise.py,sha256=bbp_F6GlCrqmGH8vvlvkcXz2_lsTATKXv84D7TyEwjE,5528
priml/baselines/craftax/game/observation.py,sha256=BX5RiG_QuzoeD46Z59ppJW_1oDyKhrd29l5HvltF87g,7791
priml/baselines/craftax/game/state.py,sha256=nXAPBkkq8UUyXn1FcfZbkSJiaEQIT-gFrKsStRVciZs,16824
priml/baselines/craftax/game/step.py,sha256=E9R4K8nZ9ePGHPz98hIrAuaL2AJH-3slnxJBZe5Tp4g,8290
priml/baselines/craftax/game/survival.py,sha256=Y0Cyh2TWQw8fh5FqByJeouvR-MYBLgPyHvKv6wOdJlg,7228
priml/baselines/craftax/game/world_config.py,sha256=5wqwaa2x7g0sWBWEKNzWUSeaBugrehlDRaZPvx7WbGk,11533
priml/baselines/craftax/game/world_gen.py,sha256=KNbc3169Bm56oOv8TlCNzsk_NkbdHj40JUPahgCY6JU,19546
priml/baselines/craftax/game/render/__init__.py,sha256=ZyPPi_t_UNmuaWpOFl4GuPxokYFKveUSfaKWyJxc9CY,449
priml/baselines/craftax/game/render/assets.py,sha256=Lw8wf2pwF1puwrLHKMtgZPJKU6XBFrR3B7LY7RIYcl8,3945
priml/baselines/craftax/game/render/pixels.py,sha256=FhSa55fmX1vFUjsoMsV1KDNDrcH-i6zuA4rW5SfzZG4,12058
priml/baselines/craftax/game/render/play.py,sha256=xOuf6oehvVSZw02L2IZ52qnjoCyJrcMUVadLJxk65EM,6674
priml/baselines/craftax/game/render/sprites.py,sha256=yZ_nbTk4YgXqlwzGEBCfoTRyzotloK9orCOUbuebaF4,3563
priml/baselines/craftax/goldens/craftax_ppo_training_v1.pt,sha256=o9dnblyQ_5jOfvgyaOniBU1Oncx5Ht17B4utgKUOkC4,33069
priml/baselines/nanochat/__init__.py,sha256=yk06ifA0bYcK_p98BQVNzyl6aKo8Vlmvua0VrA9NWKY,956
priml/baselines/nanochat/data.py,sha256=tRfySDK-N2HrZv9Tz9KGAXQEoE6JihW3xI26aEs9v84,33057
priml/baselines/nanochat/experiments.py,sha256=F1yw2j3IXdYPt03nEgpMSomRuXn_N1OMj1xPb7anAbQ,16067
priml/baselines/nanochat/flash3.py,sha256=SEmfCycGGToWu72oqc3QNga_xIAmW1cAr39rBdOs3t0,19397
priml/baselines/nanochat/model.py,sha256=MBRFbVriYQpZ_hlZ7UjtxtNTde6aG_l4FJEnyKrQTeY,20503
priml/baselines/nanochat/train_step.py,sha256=pi6PfrUBvoBtHnWJAOjctHqS2cpJhURaRVMP1bNhUo8,32011
priml/baselines/nanochat/goldens/exp_smoke_forward_min_cpu.pt,sha256=gbpHqWOMVPD8nq44MU2140_gRljJ6WDyn0PvURg-r74,111291
priml/baselines/nanochat/goldens/five_steps_min_cpu.pt,sha256=sWO71GGvMo-kBEYEIy_BzExggo46Q9e8N0sBoCFdQo8,218653
priml/baselines/nanochat/goldens/plain_min_cpu.pt,sha256=FSD8RHyCOED1Z4CbPMmOcpAxgLg03tkSnQp1u5BXaHw,39485
priml/baselines/nanochat/goldens/value_embedding_min_cpu.pt,sha256=KPpooJEU5UBfUZOynr9MYgEF1_x7Vpb66JsgUSHaamc,45089
priml/baselines/nanochat/scripts/__init__.py,sha256=CVMwJNXwkkdQl7_XcEb14D6N5tU4qjDm3HaKxZmw7a8,59
priml/baselines/nanochat/scripts/karpathy_data_parity.py,sha256=r2813X6N2FKik4WrEv8yoB0Q19lgozyQTTvc5ZSaVZE,12326
priml/baselines/nanochat/scripts/karpathy_parity.py,sha256=mDMNGc6Fhq9vNOHIbJ4YuJugNo__nHP3KEx_DoVaqms,38060
priml/baselines/nanochat/scripts/karpathy_race.py,sha256=OG9nEhWcYo9rh3EnqD22YdOd9kZ5AjsXGcGLNg0hCS4,12724
priml/baselines/nanochat/scripts/prepare_data.py,sha256=saHX4fL1O_jZGboebvpxbGvJtDE0sybdF2DVuAT2LW4,16205
priml/baselines/sudoku/README.md,sha256=Hico3DAAZn1j3Ycr_kG7x_Bg8dbNdvQ7bsEcuYyhZMY,3263
priml/baselines/sudoku/__init__.py,sha256=Zg7tgoSTkNdZigMjrbtsXpD_IgtAnWJ0Wvs2z6fKVKc,506
priml/baselines/sudoku/act.py,sha256=cTS16ypmG0BZKHvxkFHdgD2ixKptbkNBmF-ODEJS4Q4,15014
priml/baselines/sudoku/data.py,sha256=_ZYUVYnXlSQABAuios8vbpjYIiFmSz2mdoDRahFd_aQ,15941
priml/baselines/sudoku/embedding.py,sha256=Z6lMm6d3oFXFoSr1zbc9K4bItGOmG5-pdNf148pktGo,13755
priml/baselines/sudoku/experiments.py,sha256=-2u7nEkjYyuWfsIvVGT0Yf7kwB-3NMs1KSiCS4Vtuas,7626
priml/baselines/sudoku/metric.py,sha256=k3jWF9yhEenO0hWd31V7Laf60x_s9dVD6gChf1sFcOw,4639
priml/baselines/sudoku/model.py,sha256=BsmTU15m0tnT5mM6zeuq2Asu6Et94WsJgGem84pytQQ,20253
priml/baselines/sudoku/prefix.py,sha256=RU8RmtaBpU5FGWq66bpubRxLrbeUwET7syWt8yLqXMM,11621
priml/baselines/sudoku/train_step.py,sha256=--kUnnZ1HpbkBTKwMUXOAJV7AJX0q6fF2rcIH1HRzU0,19850
priml/baselines/sudoku/goldens/plain_min_cpu.pt,sha256=dzH0AweA9hFtFurnx1_86rT4Q4kU8Kv-VzLA_WlcJdU,46223
priml/baselines/sudoku/goldens/recurrent_min_cpu.pt,sha256=Etglth_rnJ8kxEa8ClRrPjrru7MS2zg-5j5vAl537i4,46407
priml/baselines/sudoku/scripts/__init__.py,sha256=zdGqs0K5nDZ8gkXjgr9lo05Ll4pfPDc7ng2JGW40tO0,57
priml/baselines/sudoku/scripts/prepare_data.py,sha256=fTRl709WHOxQnBvdnC1Ab7jxstPL-svoXSPvjTCTOx8,12259
priml/data/__init__.py,sha256=5nY6m6LIZKbxl_wi4E7YF-W6pvFKU3B3xkLH1bNcoRM,71
priml/data/augmentation_gpu.py,sha256=hceeFzLaU1yKu2kIoIXZq36XZqm7cwckpY-GLoDbnLY,3825
priml/data/custom_types.py,sha256=pGjaQs--NIFII8z17tVqOhBAWfwxvnf0d9Fj4fPau-4,1620
priml/data/dummy.py,sha256=KsNPVdw0fGHrd8VvKeJh2mFY0rpdtv1SqpgX5ek1iCM,3663
priml/data/environment.py,sha256=Hn3xZnzfrrZqthhe41qbga-utm8qgnrzqGlcixh4alw,3141
priml/distributed/README.md,sha256=hVjVh8FBFImTdSeoYrsOYnXUkIyxfVfolDEsGhKFTbQ,6040
priml/distributed/testing.py,sha256=zaDgYdqUI9bCZU6mf_joPFIhxXPz1sHE4p_kmNjQfpw,17350
priml/inference/__init__.py,sha256=iNAc0MRRcHbzfTYdG12-DEmPRGzVm80AcV6esFM4x5o,61
priml/lib/__init__.py,sha256=8gOpaeNpBc2xiI68JpniMb4xyiviMeh1ueOEvYAfakc,82
priml/lib/custom_json.py,sha256=xB7ydcEIibZdNxE0S4HowxjSbMyussnJBeyn1TWUjmE,36413
priml/lib/custom_types.py,sha256=Pu0iXtMSBT47sax2pRXBjfXaFRwX4PGJPCfKUUzliG0,1499
priml/lib/traverse.py,sha256=jogFjY5C0wv0-RgO3cWobaNZ3bKHMq9P6OG0SAKyP6U,7174
priml/lib/userdirs.py,sha256=2TvOZXk0hcT_Ez21akBG1TwmVJ7hfGJjl6fmsdmqPTE,7965
priml/lib/testing/__init__.py,sha256=X9eJS48tXcnRJ7KFDHc6CUQQipFbnPYyW2qIyhdUSCg,140
priml/lib/testing/main.py,sha256=sKBvJZC1H1ROvBVb7gLdjeaA68vM-1in6mmmEPyYdHQ,1046
priml/lib/testing/resource_markers.py,sha256=x0O4sDI_Xugw72Vj4xQrUXNy5PaUQaqcUA_bDTy8n8w,9789
priml/lib/testing/userdirs_fixture.py,sha256=-CxCqdXsaC0UjJpXepkBcf9QWG8sCWLxXtLpuDqr_dY,5887
priml/loss/__init__.py,sha256=7r26dcMz_MokQAyhL4MRyHclJsW8ZI7D8faKC-k2284,474
priml/loss/custom_types.py,sha256=0C6q_PuksL0a8IEx9H216xMfkFpxjC3wUSvBPOMkbmY,1260
priml/loss/diffusion.py,sha256=EFV-jBRO3rrf4VOqqj3rjVOw_DOYh2s-tMgucrJG1J8,5081
priml/loss/gan.py,sha256=5wGsjtxjXgghaKq3FjWzmu287CdB3jHhMkS7h6gyutQ,2292
priml/loss/lpips_loss.py,sha256=ZkRVYCxQ1XXo3V0cu-1IlKzb18y_mZcbtqb1q4Gn_BU,2358
priml/loss/policy_gradient.py,sha256=gDKdpV5LZiXeGV1f_jWIt-YYJBFT2BTzgU29ifdtq6w,4387
priml/loss/simple_loss.py,sha256=HVEr52DaJdIbCU--0htc28VOtY8JzFtcANO1foDjcOc,2319
priml/loss/weighted_loss.py,sha256=ovWXkZXr6oLjNIcTYTkF5RYVXmMQGkVxdoMv4YBiJ3c,3271
priml/math/__init__.py,sha256=6AVqEnNppp6iRlFa97eTDgRcbk6x5NywgEHaWDsFAzc,58
priml/math/activations.py,sha256=X0DzN9v198Ek1yNIdiXmJYNACxuThXm1WgYxB7NpEL0,1471
priml/math/advantage.py,sha256=9UJsULhYbqmKiFk3pvL2P5ZIjIB7rb_5_EH1GgYD7m4,5603
priml/math/basic.py,sha256=xJ_woBSFqp4cHEZEbqKZkpr-hlP-Cehk_BLvhyyxtFE,5338
priml/math/custom_types.py,sha256=RmifnWQNzJhLPHVs5dLgpzCfOLx009XMqSPLH7U-C-M,7488
priml/math/distributed.py,sha256=O9sQf50WnNYEf9PHuYKXCtgNAkLJvfA1t2AFwiv0EvE,2443
priml/math/frequency.py,sha256=jUhjRhXRX9McT1n0WgHDy_B9YXHWX7FwRTPaK3xMXXw,4348
priml/math/loss.py,sha256=hlPuIyzo-HoYsRrbTUErkzVXjao0p4DHms6i4IUZznc,5195
priml/math/numeric.py,sha256=lMWSqkJX5hV3Jp8CSwPL_jrLiOZfxnjvXMRR5SJaIUA,27193
priml/math/pixel.py,sha256=uGOWRNF7yiXm3jSNYZxBF1STaFpwV-VG40SoncHGPIQ,15597
priml/math/pooling.py,sha256=e24UbgydDo8VdeQNqQOKszT22MK628gtPU6twylHgLw,7245
priml/math/probability.py,sha256=83xqiZIsi-KQcc8WonxHXtXjHtj5InRN-SFeo25jZ2Y,19703
priml/math/schedules.py,sha256=tKkwQxTndIT1LccWnqJfHaU-LJLexr9EAfq3YB3J9ZM,10144
priml/math/seed.py,sha256=F-YkFK6ssTSzT5FR3I1YvM4YoRdPzck-ROWdD_8A7Cs,15189
priml/math/stats.py,sha256=CBXMGAf2xUxSRcYuIbChItGGjq4QmvBGiYzSgK7FINE,14289
priml/math/tensor.py,sha256=FsFF6b7WKMiqj3NU9WHgvu3pnJFN9ZZ55GEFx_IcvbA,5818
priml/math/diffusion/__init__.py,sha256=XB9qvLW06AmvCtWXgW3KXk5Qn35dOgEz8cK1WOZ10oM,2002
priml/math/diffusion/sampling.py,sha256=6wCqitC9ndS_Mk_Joi0PDeyKzdJ3_CPIjL06BzZDCxU,15047
priml/math/diffusion/schedule.py,sha256=qcqSnTTpJt5uY9EEYQAyEPD1OHofcIh7ioedRrzoWjE,13819
priml/math/diffusion/target.py,sha256=XDaeixwDJSqyuhS7tY2IxjgKtPRJHfHcqWen2sY5js0,12055
priml/metrics/__init__.py,sha256=45-BVZtModJCvEDvZKdyvdbPTr6v2Fs8NUCccXCJoJ8,342
priml/metrics/binary_accuracy.py,sha256=-kRMibf8J8sLMGl66qu3zW-AxVnGdJQRiHDW5tI3caA,2274
priml/metrics/bits_per_byte.py,sha256=0PvZccCikhvol-Qfd6-p7PNT_0lAZMI_u6UtWv3S6aQ,5430
priml/metrics/custom_types.py,sha256=DDDcsePXQsxFtfOdBw0rOg6RygHLShiLWB-jcop8goQ,1226
priml/metrics/topk.py,sha256=xUzncmM-mJbPqmWmU3I7gTss7zAykwM0Bri309FvLco,2871
priml/model/__init__.py,sha256=h65TQ4QXjbJ3vEyKCXUNDgoWVBXo6ZSOxuBZ0KouO1Y,2331
priml/model/attention.py,sha256=H0D504D0Uf5WUb9VXcVNETBGR6OXrQMZZD4ppHrXKxk,34882
priml/model/causal_lm.py,sha256=ccyh2jsES0UjLcNtN2W2TsjYm30lDl7FJoO_fDIISvk,6817
priml/model/conv.py,sha256=3-WNdcxtxlOyeFbkRnexYzAe8dDSNgn1BhmKuDAJRIE,8067
priml/model/custom_types.py,sha256=VMkLtOrcMZFwDPyuCvFcopgjiBMPe9zKHaxFgsW2c2c,7886
priml/model/embedding.py,sha256=2ukhukbbLfqGLpfPyYg06JA_WJAPN9wUY09HtHvilZk,2884
priml/model/gated_delta_net.py,sha256=GkTO9PXlPBsf-aQg4F0Qs0wVzh8fNs-vaHXPS_pmIdU,9755
priml/model/generate.py,sha256=eT-e7auZSkRE-Wb4cQem5uGikCmSjK1Qqd2tuVeHAzg,5702
priml/model/init.py,sha256=Ip__WiwEpC0P-jCcPgiZ3JhbWmcXI_7f4rQZflJSHew,4440
priml/model/kimi_k2.py,sha256=2H7WPBuWcZFD_8pPtE9rT4AFa2_uIX4n-JF_hBKvE2g,17887
priml/model/kvcache.py,sha256=mFEQMI8y0KajB_I4LpItJ_rUsVwYMPxFBCrWIlwfLKk,3822
priml/model/linear.py,sha256=dDvdnsz5XLnXGzrc1cTOO5lv6yxt8hZXILa6oPbKtmE,6877
priml/model/mla.py,sha256=phzkt0rcN9bS0PTsooklIxlmRFM3V2QKtwyY1COFs8c,26129
priml/model/mlpmixer.py,sha256=GOjpvT780iXBBXHs-Th0iFXQmmj_JZ7acd2osIbkkf4,3806
priml/model/mmdit.py,sha256=QIbZMnY7ATaGPMGesUM-PYbBOkaUqkhO5g3WWgVjvAU,9547
priml/model/moe.py,sha256=KVvZZnZtbCOIdLnHBH5B68tgWzKIWMkU1qjSgo3jjNY,17259
priml/model/narrow_embedding.py,sha256=p6VFQOKLJPIs2oAMzT71klsxX5Nj1YZqHrhXKmdj9lY,3250
priml/model/norm.py,sha256=ds19bSef97jK1DluMhIUEInvG5o7sOK_eXwfr_I74w8,15616
priml/model/patchify.py,sha256=i0E5P58W7iNxwNmTJbWRNhVWgYV1syxp-jmU5IM3shs,3923
priml/model/qwen3.py,sha256=4LCg_91YFLAN0HjRnG17uJP8Og1f6Aeu_xXa49R72IY,11097
priml/model/residual_mix.py,sha256=N6BVrgitsg6FiyOaeJWM1zg6445H1LCQC8zODKiF3zI,3062
priml/model/rope.py,sha256=4gBgOd--h8TAgquptdd3bk9CoI1occhj3phIbhBooF8,26591
priml/model/sequential.py,sha256=rOi9Tbyw2CTp2Ffpk3yk2V--utwTlHH9z651QG-kQb4,3927
priml/model/softcap.py,sha256=6Zfz2ln5EeoYw1HB2XVuyp72Zm7TEa6G3Kut68A3Stg,3134
priml/model/special.py,sha256=5VQeUdTa6WGoOgqwdUV9DPHap1TAkzJuDHFQXNxaVgY,1889
priml/model/swiglu.py,sha256=4-NDHgIjgTfTitgTali5yLXXmaHFgA_zfeon1zbX4gM,10469
priml/model/transformer.py,sha256=oUf8KeJ2JMl2wIi1ZQ7mI6OqJjQrMpTXZ7Del_sxAVQ,6532
priml/model/value_gated_attention.py,sha256=t_3JmFiFfXQRuinQSerME2pFPUnqym7rQmBxt7kiTqc,12782
priml/model/whitening.py,sha256=QRI15h56VRHZt58AEn0oFp0avt6zW5GEgY2Lhd4gMZ4,2087
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=IfbMIEc73S02fCAD0SSFl0J1LAROCV2im3Fw0OeNlJ8,4722
priml/optimizers/adam_atan2.py,sha256=hTIyvdmRYiBYzWKdl39Qhj0DsqoDHtDSWKjAG65Dy1o,4996
priml/optimizers/composite.py,sha256=JHNTFyH8_hpwD2WztjKnXZZsm7onKZIPGUgY1N1qn7k,17333
priml/optimizers/fused_adamw.py,sha256=Ob4FSslzW1N_WS9Pcv7wTgjOIwhkDRTwxVy4ChGcMS8,8702
priml/optimizers/muon.py,sha256=SV9MHix9G-ySgetEF6EGDLCeDbCE8nO_sA5VwOBDy9g,8927
priml/optimizers/newton.py,sha256=oTmYKmKEt6obgG3VgXmtimLvlHemQPpHUQgFbbsYrUA,8112
priml/optimizers/normuon.py,sha256=UU9yYna-KoGQWBcWGdGmtaibpflKr_thtIT0pnYuvio,17134
priml/optimizers/sign_sgd.py,sha256=sNr3SNmeoOSODDFO3UpsJ7pQxyTEIPA8PmriTjQif4I,12687
priml/testing/__init__.py,sha256=EuGRevcV68b4POes0wZugnbhQ_TXUR_mI-XJUQXCq5I,44
priml/testing/bfb.py,sha256=k248LQvACjw85asFVi7YOkS8d5Lx0Hehs-ev02iFk8U,40110
priml/testing/fixtures.py,sha256=DBmE2BRb0uqCBWpsTnm7jEPtroC7JsSjIlygt9CUuW8,3927
priml/train/__init__.py,sha256=bRGlUC8hD5Kq9fP3_kSRwZndkohpLhx6N_7VUeMF5eY,1788
priml/train/activation.py,sha256=mYajo8Gc-z3XsdvqHXEIAlyDViJbrQFqtahEc6tBQLc,18435
priml/train/checkpointing.py,sha256=eZewBBDhAPWaOdU-eA92bNltT4OUJNfASSV22X8zcPk,33119
priml/train/custom_types.py,sha256=Sr5eMERWL9MsKcjCINuVbvtZX16Mb8ks_BXBOep6r7Y,15064
priml/train/ema.py,sha256=hzQM2BUkjTkyfo3KHRpMIuCbOpoRVLjzIGXjf1AnK8I,22441
priml/train/grad_clip.py,sha256=1PpX4prWQKfonP-kEIsaGszWzNnzshR58cIyJrAet-s,5551
priml/train/parallelism.py,sha256=xNEdsjPOesDcNYsyzUz0ZgXRiHIvy7td_XKh2p-Ituc,20805
priml/train/profiling.py,sha256=6PTdc5RhKSoKenEAOs4gr-DXgdpkVhgP_YLKrt2v5Fw,21061
priml/train/progress.py,sha256=OUKphCFiMay5jeefmeRTR4MwsxpqyfXNZoisVbUhX9A,1927
priml/train/quantization.py,sha256=h-mllP_xbVakUN80s8mPvC5_Ed1GGq87PJK2BCk-J0E,5890
priml/train/results.py,sha256=DFcGXOzZ9T291KvXxGqIAzKd6hMZBw9W4cidKVLLOb8,3020
priml/train/tensor_parallel.py,sha256=nVfJHvYbq37S6o9PQ1a0XyOkdtgKwJYWBpyCsVaNkEI,8231
priml/train/tracker.py,sha256=nRlLfNvHw-5H71ep4esTuijSJ6kl9yVNcZomP72xgd4,25849
priml/train/train_loop.py,sha256=DwA2QiNWOyV5mbeaITmp1ITIKzq2sXj6sgEegzFCpps,70393
priml/train/train_step.py,sha256=NKTSF51rhjL8E-XclT1tWR_ZJfK6Y-pItrPMZ3wRnAI,35732
priml/train/train_step_gan.py,sha256=vxZLPPqU4O2kCb2qJdRz6W9CS5YTboe9rvM1Nf0zr7Q,11790
priml/train/docs/mesh_native_design.md,sha256=-9oMDGsRs4E0DRWI-4RPZ425mAPO0avKpVCRWfszmqg,10586
priml-0.1.4.dist-info/METADATA,sha256=dHh-un5cHNE14FDrxTsjwZ08Q3opuu8doEKxladG3OY,4976
priml-0.1.4.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
priml-0.1.4.dist-info/entry_points.txt,sha256=fQjHI0uD2IsG9qa0OhrHCoDnCLJEiyP9KAd2htSytEA,44
priml-0.1.4.dist-info/licenses/LICENSE,sha256=UOZ1F5fFDe3XXvG4oNnkL1-Ecun7zpHzRxjp-XsMeAo,11324
priml-0.1.4.dist-info/RECORD,,
