trax/__init__.py,sha256=-uOEX8IFahSc5_GOQ204hTkEx7MNxnnGjuPvroewgCE,850
trax/import_test.py,sha256=6j9AvNKG_0DqAnF-y81LhlPXLajXYMeDgOCl7QThefE,1030
trax/jaxboard.py,sha256=5i2DVJUc2sGg_Anru8vu3otygX9PfjWghNa1iYpgwAQ,11661
trax/rl_trainer.py,sha256=W6srvItKoFrjnF-e-oeNPC_3jzCXx-Jo2nGaM6lrIqQ,7288
trax/shapes.py,sha256=5HvSoryKexGFdDs4L9J-6tPp7a9pRsDO8hOOngetCwI,4422
trax/shapes_test.py,sha256=npZPfhwYFdZbUQAPuOpx9G3ZIzVzPQWGtpAiMxDVFSc,2743
trax/test_utils.py,sha256=n4C_Tg9b3WnYODhAGEPF23C-zWaFpH9uBOoE2BISJd0,1093
trax/trainer.py,sha256=gQE1bAqlMByMVOTsyYNL1v8OBdBdeJm5iNLg5jwxn7A,5680
trax/trainer_flags.py,sha256=tkqmYuKPSwYY4l0E3ZK_GXftLD5GUmVAt5ObJZDOw9I,3647
trax/trax2keras.py,sha256=p9iPmHin3VvNzLaeYiQBUGBKVFZTzMmnRL3SntEJw-k,8043
trax/trax2keras_test.py,sha256=bIqK6jazKcyMnNsiAHgwoKvXIU_SNSSUJyLF84NicLE,7502
trax/data/__init__.py,sha256=ol_Bl7NuQwh2CkVQ1w4RxuCfNnKtsuTOSNBO0lQ9_tA,1493
trax/data/inputs.py,sha256=Av7VzdN7VAejaAQW2EDYjfvYUHSyAviA5gFA5jirEow,30990
trax/data/inputs_test.py,sha256=fdQA_rUX-l0N73jz6DEPLlmAFwtZ3PRe5YXIPVP2JyM,5947
trax/data/tf_inputs.py,sha256=DzcLS0OITsynxxUNDV0WYJVvscm4wp75kjER2TN-6a4,36681
trax/data/tf_inputs_test.py,sha256=H2BlfYDPMxURkxUmRDSBSR89AAo1gqBoEshEEA4xFzA,17964
trax/fastmath/__init__.py,sha256=HNRFFvwuiIYev542WSX-fyB5l1TDAot5CKsy0HKVpMA,1156
trax/fastmath/jax.py,sha256=Zp98j3GE-Qe01Ol2DAHANUHc6pc_qROLJMCc7xvtwx4,14711
trax/fastmath/numpy.py,sha256=7kwXAQ_oVa8yE2GCfXv2PVuuK13PPQpGsbbJEiDgVGk,1241
trax/fastmath/ops.py,sha256=VpHJzhcTUn7ZdA7E8wNrzh5hbtLNcKRwo0DIQda0ic4,7518
trax/fastmath/ops_test.py,sha256=Qfjp83bXrxDvKVJLl_H4LSVdgd3VWsLpGLRRPXzDGK4,2774
trax/fastmath/tf.py,sha256=lmSNgmTVpaQ2jovL20w230U8HgQqHPdMUqdzRdua5Ug,4897
trax/layers/__init__.py,sha256=gUjh4XjHATAEtxnbTgPqsJn1x78AQdbmOnl9gQmJbKU,2769
trax/layers/acceleration.py,sha256=9MOd02e1jZnxsUcIcUgPPPmof1f8c2CUo0xhpxq_yjk,8868
trax/layers/acceleration_test.py,sha256=B9gwlrwBjKQU3cy-Jdfg7pGYPqJqCiN9O5fhhMG0E9k,1708
trax/layers/activation_fns.py,sha256=f_HE-xzG_vdhNA3_Z_FipEcwgV-rMM6k-b58YZV16tM,5939
trax/layers/activation_fns_test.py,sha256=AQ1doBlpJuEth4vXZ4FerLtaWTPSU1vgNkifpBAM0sQ,1833
trax/layers/attention.py,sha256=bJwTSazo3a__1OU71jiYQb5ld7ai6TRaVybfmvXTfBs,17599
trax/layers/attention_test.py,sha256=HGB72ihaLim_pTXbUYzYvBb6T0_aP8wrJnwcO_JNx7k,3025
trax/layers/base.py,sha256=uNiTXUAnShbgmFgXpB67okPdK-jixaCVDIAaqX58DgM,35301
trax/layers/base_test.py,sha256=Yg-CqzLxD3enWvEJZuJLKgIlSicTOAxTCXb1i-93fh4,7052
trax/layers/combinators.py,sha256=-3V9g2cC-RCPOpu3YAJjIiGvpLA-oCq_d49AbZdV-V4,27409
trax/layers/combinators_test.py,sha256=SIoU4vax0kdX8i7h0jqd9OQwo6ZF34tYFiQAVSt9VXU,19359
trax/layers/convolution.py,sha256=-cMT1mgvDQK3tQX_czA3g1bcClqqHCOgqI2AatCF-rY,4924
trax/layers/convolution_test.py,sha256=Faf6jxwjG_SBwQ72Ct9RVev6wB8dzWVAsaClTLCeWQ0,1574
trax/layers/core.py,sha256=65u6kPFHM50zHCX6hZdTAZ6yDVy-V6EIfU4Rlgujed8,14099
trax/layers/core_test.py,sha256=jBx-EECSCBdLE4YXfVDfVXa8vgEeqE8sszhQ_4NTqCs,8594
trax/layers/initializers.py,sha256=z5GWpFgrIsFBU57bFNVohXJpRVXA6qqFcmDKAOyPAx8,6526
trax/layers/initializers_test.py,sha256=mTGjLlraqIcpPVRspZlIBzaWP1TxSlAgqcV78GdkwBg,3087
trax/layers/metrics.py,sha256=noDJ8CH1oMCRW4u30liZM7RG5l8Yth976e096JvxJXo,7115
trax/layers/metrics_test.py,sha256=X76wKKto1Dlm6yEggGG8C0gA8L7rKyrUk4ai9SxRC0g,4106
trax/layers/normalization.py,sha256=uIlayj9wP1i7KdS_4wfSH8NJu7ClxrV09a5OrQF-exU,6731
trax/layers/normalization_test.py,sha256=hzo--LOYC_wdAkQSZJFhgyJ7_s_BB70ioyMSTVscp1U,4098
trax/layers/pooling.py,sha256=XfZv4KEV89Cw-xzBD4QCk4Jfz78KeEilT98OWivlqHg,5493
trax/layers/pooling_test.py,sha256=TNVKWersNmuNCah7oSK6Vt2RRv1vNewXTQqQGfeXxiw,4230
trax/layers/reversible.py,sha256=ap-D1IHWSaR2xx_VqEC-verKNlZeHVSZnPbgO7ISJhg,5496
trax/layers/reversible_test.py,sha256=BnfsmymohHMu9KrICMXXWcN1WkwAsOX5q3nUw_a0shM,1249
trax/layers/rnn.py,sha256=XD0eefIZ2Jmafmc0jTd8Y2Uprh6-jSfXxCbJ0Uejy48,10909
trax/layers/rnn_test.py,sha256=88Kt2TE4fSjFLfe-tSZKuI34AJQCp3hnzPw-jFIGo1s,1895
trax/layers/research/__init__.py,sha256=35TrDphYwu0MVtq7R0Eemk84z9EorpmyvWF4vQPOjsk,597
trax/layers/research/efficient_attention.py,sha256=D7OFgE31WAQOs5H_-EWTa-k463WjcdMgOjLADrJttOA,60975
trax/layers/research/efficient_attention_test.py,sha256=OqmGgbJtcVb8hSOHdhQUAVvFoNDuPKvtfuk2GST0Z08,11575
trax/layers/research/position_encodings.py,sha256=qgDtIjjYv3XLswCiDTKaKXj3HWF6y3MN_GaUaLJa9lM,16186
trax/layers/research/position_encodings_test.py,sha256=Av8LbiE_Dd6jWc6h83jmb4w7u_zi7bOMoMZcgpMmNGo,2880
trax/models/__init__.py,sha256=9XMnD_TDH_eRVhqhpmubd0tP0vjVhwL2wOt5urycvSE,2612
trax/models/atari_cnn.py,sha256=TxmSLyhuurCwMXDhZk38ehPwj7PoP3-oqM6abU0PQHo,3276
trax/models/atari_cnn_test.py,sha256=EYWT2jl0bC4WU0MiI_mbufSPL0R6K3CNkpquI_ULXkM,1859
trax/models/mlp.py,sha256=jD635lW65v3A2XFfk0nVpobA1w2RoMWxGT7QvE2y4Rs,3119
trax/models/mlp_test.py,sha256=ef1cQDzbecdoVIQwf-MBKCwDpyGiTdu6na9TZ9OYLqY,1336
trax/models/neural_gpu.py,sha256=4GvRZKO1SKyOJk6opgPLia2GgMqIFSUYARmTJTJtjJ0,2565
trax/models/neural_gpu_test.py,sha256=a92O_PavxxiMXV2TlaZT8oSQPXDXs-W8k5Tcy6bdfTw,1103
trax/models/resnet.py,sha256=t2fv0BP2QjhbigkeNfUA6chXbmBFpu-PNMcgSDMYu1E,5959
trax/models/resnet_test.py,sha256=f6Vh8HnKHdjYBD-7RDEhCkiFDAITx8awxGxib5kTpOQ,1353
trax/models/rl.py,sha256=2OUwEyqKk5FAaCtghTCeyz-o86PQb4ZldDfLRRh-ZJg,3531
trax/models/rl_test.py,sha256=7QLmUzpx5ajdsXp1Omwrm2SCiUr5IgxEg0Zc5ZTX8s4,1627
trax/models/rnn.py,sha256=igqDTDolUjG1AmimGsddUqr6qOsNKIYIO4kdByckUMU,6229
trax/models/rnn_test.py,sha256=m91wiTQi0cog1PLs-YMTWwU5mvAet6Rhq0IeUmSWr2Y,1294
trax/models/transformer.py,sha256=jaNyBMwgEAlnENH4W6HjdWRgiFEACEmf5wBqCWuMsxI,22431
trax/models/transformer_test.py,sha256=1F2tI_H09-4dDbJjwS_qltBpz5dfrcHFkhrXe3sHc7U,9608
trax/models/reformer/__init__.py,sha256=35TrDphYwu0MVtq7R0Eemk84z9EorpmyvWF4vQPOjsk,597
trax/models/reformer/reformer.py,sha256=8ClivhFwVmdoEQ-Ts6kpC2-H6r1S8QzTpvFesmeo-84,36523
trax/models/reformer/reformer_e2e_test.py,sha256=yflo2SWR3kCZ9zw_YdMHW8Lr87i3bSqOJDUfj6yb-uE,2892
trax/models/reformer/reformer_memory_test.py,sha256=M0NtjW83I1O1b1VA0RVNgt367tdjb6KaH_TzXIBDLKc,3060
trax/models/reformer/reformer_oom_test.py,sha256=fcV-6d4X6mTeueXXaIKGRHnNYt9yygS6moS5FIhLoX0,4058
trax/models/reformer/reformer_test.py,sha256=ZJ8gSjUrHqH8FU9vYqswJ_DLdrcCseE1Nfawjflo4dw,5651
trax/models/research/__init__.py,sha256=35TrDphYwu0MVtq7R0Eemk84z9EorpmyvWF4vQPOjsk,597
trax/models/research/bert.py,sha256=JkcHgphLNE4baJNwR7Ui73AaVkGT_QHhe0SajXww1Zs,6675
trax/models/research/skipping_transformer.py,sha256=iGDhkxzsDo3IJ-fRazmMM9gDXa3zCGLWHgfCa8uCVc0,7478
trax/models/research/skipping_transformer_test.py,sha256=xai5FpvZHLF2nOrm4nkIDxifGzDQbCaoifhwVeSiojU,1379
trax/optimizers/__init__.py,sha256=IBXMJ-5R-ygol-PZjGteXCtfqlgJyNhN3P0q9AqLM60,1306
trax/optimizers/adafactor.py,sha256=a1WD2ap_rEkb6qpYSydtszSDJPaQkkkcPJIT2h6aEyc,5480
trax/optimizers/adam.py,sha256=z8JvjmSRkIELmPL4WyY7dnM_IcqmjTBIoOPqTvsd-xs,3320
trax/optimizers/base.py,sha256=yq-oYKPbzlLmoO3eIt9HcTifNduZFjE0hznd9jutqsk,7914
trax/optimizers/momentum.py,sha256=7mgIMHvBd_qcwir9uQpEc0PjR7oP9aTmgbFygofz-UM,2652
trax/optimizers/optimizers_test.py,sha256=Yu688MNy66NX_xvQ6ZPVN57WLKj_3XkIonaTi5cci9E,1533
trax/optimizers/rms_prop.py,sha256=AYr6uczd6UBAMV9L7HQwpTgVz2qR8Un6EDFbSMm31BM,1692
trax/optimizers/sm3.py,sha256=-jGvJ_f2O4wZT89-oHohicE0QdiRxScAsg8LIf7ePDc,3744
trax/rl/__init__.py,sha256=0LZdDPgisvg-bXzuak2TdANzLIMkjhwsqRbYOKokstw,1754
trax/rl/actor_critic.py,sha256=lSySrNIcJhKlsm_9kSMbtAy5N-NjpQvviLEPTWmBnTY,33889
trax/rl/actor_critic_joint.py,sha256=fcSTWG7IB4KbIe9w093eRD1lFVVzxVQm1oUnEJIWwDk,26136
trax/rl/actor_critic_joint_test.py,sha256=i27T11HfAiz3H20eCnlraMA0J3KzzKig4cpaNV38T6E,6153
trax/rl/actor_critic_test.py,sha256=Hqk1SS-e0WolaTCqZzRT-NiZBUAIucVOeVCnJ9iO2p8,13262
trax/rl/advantages.py,sha256=xXAeiZ3ZRTsFmywjo5myxqPYC939c5O06_BipC7cbRQ,5577
trax/rl/advantages_test.py,sha256=gzntexuJyaFykIYnFkqIo1svs-H1s4hZC5txwwuWmSU,7384
trax/rl/atari_test.py,sha256=EjpnN1UN4lTYgMsGoUXaMg-GWzK-AXns46W3mQyPZ6A,951
trax/rl/distributions.py,sha256=_T6yohHOSL8CC19hIRNUTRF-YOvORDjB4GHs9IOIFYg,6293
trax/rl/distributions_test.py,sha256=Ff6ISfE-hm-Y9ko4Xv_y4U3Vk01H0uobmxa5uVHmCms,2467
trax/rl/normalization.py,sha256=Ek4UdALM-ZaSC72_UX8v63kPzWWNFqz-r9neqk1Vs1s,3461
trax/rl/normalization_test.py,sha256=IfIriEU3EizaSj2N0t9-uU2SK5pQgFICbCyKOPo1yG4,2333
trax/rl/rl_layers.py,sha256=tEt4uIps8KbioV7C-whQkHib27wRDqhzvFX1q87KiuE,9323
trax/rl/serialization_utils.py,sha256=FTBppoXD-tCXWhwbPS3bvfUrhiXdPXMMZbRJ3ec-DHo,13938
trax/rl/serialization_utils_test.py,sha256=isG7QyZgqVmUyFJXHM-Puhx3dDB9UePpFXiWcO9xLB8,8452
trax/rl/space_serializer.py,sha256=34mB5T-CVjFiUuZ0ihevM_IDslRp-VgmBW5wZ0eKqV4,7016
trax/rl/space_serializer_test.py,sha256=DbFpApALDrQQL2OVqFwbMtWe8By6AGpEhjxNQRrgFtE,5965
trax/rl/task.py,sha256=fZI4AUH1BGhqS9PI8f4yxICRTu8wgrYcstXh37elgCs,27218
trax/rl/task_test.py,sha256=9bleRKcoUGgP8YRHSftbsdtKROtMcLq5VxcN0-Roi88,8065
trax/rl/training.py,sha256=YrvN-DoTrI_zPxh7PRnMYftDdu2AgZS2z9x_HhWUQgE,17164
trax/rl/training_test.py,sha256=iKiDBXjLgRRpXXTv1MMrVneYOLCiXAwzaZbKNuSBwpo,5233
trax/supervised/__init__.py,sha256=VOc5sHveaXGCioZk0s-uFwqoDoyQmCJU__5TtODLNzI,983
trax/supervised/decoding.py,sha256=3MsQt2O8Bwu7OR9FAHZiFq9fvGpbivnXS4pgt9PHaME,5117
trax/supervised/decoding_test.py,sha256=1WKnGuGfAxG5fRKyyL8FLy9TZq4pDOmUyTXcYEExwh4,6946
trax/supervised/history.py,sha256=S9-x44unXtVlRCgDCArmshsEZvsTEIdHkoviM0GcYbU,2295
trax/supervised/lr_schedules.py,sha256=SQX-y-VUPmNGvyKMeMlLeLR4wY23Xwer64ibx3WJt0s,8324
trax/supervised/lr_schedules_test.py,sha256=N3Eq_fc7YUkYRlwYLSzmEu95ynutp0VfoTGJLgd5bL8,3304
trax/supervised/mnist_test.py,sha256=4_jVa4iQuxFST0nsuXz7zSetPIYo1x_2AvFLtaJVmlw,2305
trax/supervised/trainer_lib.py,sha256=s8yjwj6GFj1BLS-HHFnezn_KJGKbPWpvEdgmir_kQzk,33929
trax/supervised/trainer_lib_test.py,sha256=WHnQ7H5FzH6BuYD5OIvQnpumcHLlhN4j-oJfHZ5Tj40,10639
trax/supervised/training.py,sha256=kds5MpWnO3-K4XcJPIcgqL2nZI56lHd1St5M4cvvkhY,36998
trax/supervised/training_test.py,sha256=-Nln8NP7Av8m_9mHWl8mSoIsNRkixGMlx6w7FrPOAXA,6662
trax/tf_numpy/__init__.py,sha256=35TrDphYwu0MVtq7R0Eemk84z9EorpmyvWF4vQPOjsk,597
trax/tf_numpy/public_symbol_test.py,sha256=xJJhy8ZRC10ABWBKLhA-rzJLlBIYAiSBXG6w3nfeDLc,1241
trax/tf_numpy/extensions/__init__.py,sha256=UGSSk--UH6Vw6A3QMBageQun1BiSOtIGRVzdaR4d9VA,894
trax/tf_numpy/extensions/extensions.py,sha256=g668ATYbLptQzDAO2_I7Ldlu4S_jB_Kdq-ycAkY8gAk,43337
trax/tf_numpy/extensions/extensions_test.py,sha256=onZbsd03kP5y5kBHXnndwhAeT7z7r4jDf5N39AywWcM,23626
trax/tf_numpy/numpy/__init__.py,sha256=oozymMJ5PLm2w0qJNjYXhTA8yMLwJ_3CM275FSqLhLo,2510
trax/tf_numpy/numpy_impl/__init__.py,sha256=XVvWe6P57L5KulBPECXYurmYeuy_jhHbstbvF-qlMrw,736
trax/tf_numpy/numpy_impl/array_ops.py,sha256=C6FoM8CpNu0C54OCqvO8ozX4Uvad-bhshY_a1XGIT4c,48554
trax/tf_numpy/numpy_impl/arrays.py,sha256=7tnxvHxpwTNF9vyMwOpiYFEBVSA7aTWw2Dk9698U0Us,8814
trax/tf_numpy/numpy_impl/dtypes.py,sha256=4waG4BgOdksyqUj47TG8dlF12bXd6TFq4DZmVDtgHh8,2452
trax/tf_numpy/numpy_impl/math_ops.py,sha256=lJUBWrL6b--A-_t6sYbxs9DZZrtA_2FyvbZ7E_ZorDc,31678
trax/tf_numpy/numpy_impl/random.py,sha256=D6F0u35C_q30xkm98ZReO5tgMBvMmxrHRk7m11L1kRE,1363
trax/tf_numpy/numpy_impl/utils.py,sha256=dZL-u2WdU0JHcx4LqaUxiZEs51S2NB9A60roFrI2Wmc,11085
trax-1.3.4.dist-info/AUTHORS,sha256=EkL4K1EUyZJiXwm-BZMC7gHs84ylsRyl1fhCjf-xNLI,294
trax-1.3.4.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
trax-1.3.4.dist-info/METADATA,sha256=d3zwCYazCquTZXNGSNHtE55NNyEPi9sKshBTz6WNHjk,1607
trax-1.3.4.dist-info/WHEEL,sha256=HX-v9-noUkyUoxyZ1PMSuS7auUxDAR4VBdoYLqD0xws,110
trax-1.3.4.dist-info/top_level.txt,sha256=2rJyjFdQ3Q0lojsugL6toyALj0yhMAGuR9JkF9ElfgU,5
trax-1.3.4.dist-info/RECORD,,
