trax/__init__.py,sha256=D7byybn15NZq-jSfJvt_mmCF5RkqkxRImJuMbcRpsEU,790
trax/history.py,sha256=6Toquh01qH5WTAA3I0DnIYd3Ns_WIb1tdsNMP_MldJk,2295
trax/import_test.py,sha256=1XiAn9hJm-xEwm5KAIvpAf8C03WTEy-GBeLVzeqJqtc,1136
trax/jaxboard.py,sha256=WqPcryfgFnAYEhCt7KIainFCaqfEOExjthJoOlfMU70,11516
trax/learning_rate.py,sha256=Z9qyyya3bbSBWHI799B2kFw2cDIjWpDPNBf9Rqjbuts,11013
trax/learning_rate_test.py,sha256=WnMbaFHYjWW_PEPo96aGZOt5FiVm_DI6h7OGV3_bfGw,5104
trax/math.py,sha256=9AHR1ISNPX6UZiHXJPF3UE8egGMM_-bd02zqT3cmGzk,15212
trax/rl_trainer.py,sha256=ySkscCCXYhnfznqFRVyxpmWQyscAu2C5aWxhBSS9jyU,6534
trax/shapes.py,sha256=6CiRBf-zCl1EoryHZ5vnIjopA0u7q_3tNYIfpLG1Jec,2761
trax/shapes_test.py,sha256=m3EscVtp5JBRx_B9vzs4S1kUmHSbDeKIrxGHReB1ieE,1750
trax/trainer.py,sha256=sXGCDuMDMKu9z__dAT3A9AiDpCTpy6kZtaLYBI2Nims,7067
trax/utils.py,sha256=bwtdf9FoMD38pdZwjS8qV2KIakfjjJDIuXPtxCxPy_E,1343
trax/layers/__init__.py,sha256=YPH0F3Z7p_Q3PtK3JWRhIlFTSKdJdWsuccY11BCLgeM,2601
trax/layers/attention.py,sha256=_Gvu0g01h1APGLx_IFsbd4FR6_71YBKPBwgW_50y5DY,21653
trax/layers/attention_test.py,sha256=xC4BnCZM5guM21vsqJ3vF-8AL-IGFK1iczwXFqJrai4,2243
trax/layers/base.py,sha256=S9AMxz11JjUU1mgK71t9tONRfP4wDvpXdS68LByQvSg,30830
trax/layers/base_test.py,sha256=d5WZT0P4iYa3HgwxxrcnL9OyuOs2Ihre7W0JNPC_3ck,6192
trax/layers/combinators.py,sha256=S7iIIFvRoTywZk00i_xCctE2UMv1WAgU125MILj1RHU,23967
trax/layers/combinators_test.py,sha256=co2dX_kic6r5y-kk2wBBSW_xY1A4Dx2LvI7yT40SmIg,9849
trax/layers/convolution.py,sha256=XU97CZHalvTHTjJCXCOAmhxU_5fQxFc7w_2ztVMnIOc,4573
trax/layers/convolution_test.py,sha256=zGNlSTUYAvXgR4a0JjLjSpSZubVKyX05UIJinEsddm8,1901
trax/layers/core.py,sha256=Ew7JIgbaIJEwhK8IlKjmv3cIfUwpfnkCv7x-R-a0wkQ,8153
trax/layers/core_test.py,sha256=FDQLJaLDmF3S3mqGcF4XfKL5u1bUlQLL95Ba8fxP_JA,4751
trax/layers/initializers.py,sha256=fn0dewlDysaiyrescRBE2et5hewu9LkxXlBJKiJvAug,5966
trax/layers/initializers_test.py,sha256=0fpEChrAYci6V7L8emGC8EAGZNz4_mDvenAhq7AEe6c,3574
trax/layers/metrics.py,sha256=kYubbwgByxnRlAmZpefsuFXGGX9U0x029AbYFi0MBsE,4320
trax/layers/metrics_test.py,sha256=sT7B6Y5xtvoA-Ezxs9qWOQUGpXTZQkeOOTbPUoo337Q,3389
trax/layers/normalization.py,sha256=HdfB_45qckP9hjLvfnHVRmDsDYl2HhE39Ep3_LVVUgY,6637
trax/layers/normalization_test.py,sha256=mOueswyujQYbnMJHuAroc38IkEeoE8hXtkaxPCr8Kwk,2976
trax/layers/pooling.py,sha256=b_erJE4ils00dj9KhkyzfxDWbG3nOu1LRNAcx4nJVXA,1440
trax/layers/pooling_test.py,sha256=yaiHW1wBeKn09rKHWMZh1Fb95ZqSSUOj87w3VZeYA7I,1209
trax/layers/reversible.py,sha256=hKEgoyW1ToebqGI_e6QygyvEH48gXp0z9gLSt3I4Apg,4803
trax/layers/reversible_test.py,sha256=P9i29urUtEcn6mQ29VL8Dr_fQBPIc4FLh6876e9_gBE,1230
trax/layers/rnn.py,sha256=xsHA8YCpmQKRt341fJd_kOhAFj8zwOmHPe0FfaBivvA,9960
trax/layers/rnn_test.py,sha256=18pxL7Oi-hi_QSQd-wy_vGnsvN3ITQnBNlKWKZ8JB-U,1857
trax/layers/tracer.py,sha256=nsRmQbzmOEnxLoaR35gLC0UPf7nOG1rAobbj_PM8z3c,15477
trax/layers/tracer_test.py,sha256=2ucJvRXftu_F0D3L1nPSQ4oq9JdO3D3nimspQJGDU-g,9696
trax/layers/research/__init__.py,sha256=ggQWl-jKWH3ckyEmC2ZeGRdXStObvqYv0NJFBnpr4E4,597
trax/layers/research/efficient_attention.py,sha256=y2jtrNGPvNuVdC1Tg1q-aVtKtIPLJUde58sHMk8kHwU,45604
trax/layers/research/efficient_attention_test.py,sha256=a1UOQxJmtyArvjHogqZjcMzEyTFo2zTaJW09a0ns-rw,3140
trax/math/__init__.py,sha256=5bTqiVrdobQ3_UEkLpLovfXNHgh6bruspG1aLS_oy1A,835
trax/math/backend.py,sha256=djDFWGSU0bhnoSzqFsBpRJkE-QRvBq8aSwJFYwSFvOE,4330
trax/math/backend_test.py,sha256=Bp4fi11Yp2REOEu9wVUmKDxEvBAgOMJ4Fs82WQXjTjk,2262
trax/math/jax.py,sha256=hjwaMT1DWVUeC-ihmgzr4qhIr85Iz1y5kA8hGffX8H4,7771
trax/math/numpy.py,sha256=dEGlNH6J_QYF1qR3zhPNIDbpUOLLpzW22RXpij-yIwc,992
trax/math/tf.py,sha256=yZD27b1hQGFQwPegQHKvcPAAErEeJ9ww_iKlgjeswAE,4459
trax/models/__init__.py,sha256=HQzHlhS1YJjGWlVWtSXU5ANRc4IRc0YvHzZ4J0V_5fs,2246
trax/models/atari_cnn.py,sha256=XQQZZie-exQwA78BYmrFGLgL_z8K4uSZkkaIUefCFPE,2361
trax/models/atari_cnn_test.py,sha256=pfiVCakRlxJ3dey82YhuBpjiamvT6J0-q2YdoyTjybg,1962
trax/models/mlp.py,sha256=Xfy3dpqr99xbKavCthVz6xNxGklEc3GGznUj0QvdC6k,1131
trax/models/mlp_test.py,sha256=mUzlcjfVMsYbf5OYQ0PEomIvtakIzQy-sU4MyVHKrps,1786
trax/models/neural_gpu.py,sha256=OHtpWCdDwMsXD1oMGgXAXJ7jWz4b8ItJZ_XLXPAa87s,2575
trax/models/neural_gpu_test.py,sha256=AE8Q9NvSOvrNIJXBfJY9fknhxBT4n6f-YRNW1uBdNwM,1293
trax/models/resnet.py,sha256=ExPkfEYuURP-wxwv0ilBIMhqJ4JL6__wjufxBVICtm8,6050
trax/models/resnet_test.py,sha256=E7i_ezTekWYLTiq3yrBaIt3Ij4TnQ5beZYDY4lMvR4I,1479
trax/models/rnn.py,sha256=yEuQH3pVMRmDFjgptGWO2by0m3Fx-EubTFOjkcN4C-0,2330
trax/models/rnn_test.py,sha256=9kKWycFvazuhY34v3hGzAKe1PStHGDxqXPwHC9NVzjo,1289
trax/models/transformer.py,sha256=E6GJwcWYfDgot23brWyzNbp53zESVqzpD1ZdBmtjEtA,16470
trax/models/transformer_test.py,sha256=H0c5NXnxDlT_UIClYA-tyuCZTpdhIWT0Wd7lA5M_DKI,4014
trax/models/research/__init__.py,sha256=ggQWl-jKWH3ckyEmC2ZeGRdXStObvqYv0NJFBnpr4E4,597
trax/models/research/position_lookup_transformer.py,sha256=wYWG1MSRWxtObDcRtgWbO2Gw47JFefvEMffk8f6N830,10934
trax/models/research/reformer.py,sha256=DS3CtZrpmap1i0PIV4H4_A8tb6OSqXUXdMZxPQW0gME,28028
trax/models/research/reformer_test.py,sha256=bTaGp3hTxXnPer-1vJn9_IsMBH5DNB5iaJTMLv_aSiY,3961
trax/models/research/skipping_transformer.py,sha256=yvVHblOSMVh_GS_JBm5q95xIUUgkSE239D8DYRqOf_M,5783
trax/optimizers/__init__.py,sha256=O1Tpoem9v2iMPjUaP5vZP5vTP9CTNOHBup5DxXR9ugs,1418
trax/optimizers/adafactor.py,sha256=TJoo2fFHRGOYjjGvzGN5dgz8IE2GoGwbedzqhS6R5iw,5512
trax/optimizers/adam.py,sha256=48lJ315gJQ3Ae_6tbeXWWG40cECUuUr8Oa5XijoqKyk,2523
trax/optimizers/base.py,sha256=ALnyU_aDu_XyG5uM1AQDcrSGOw6LsKGfZHPrBZYdggw,5344
trax/optimizers/momentum.py,sha256=bFwTJAIVxvslJ-tlwpMkJlUF-scxpYvhzQTcdVjBmGI,2498
trax/optimizers/rms_prop.py,sha256=6DM-rtw1awVVKu0rk0HvfqRYMIyEm0pjOt7Gg-RgVU8,1571
trax/optimizers/sgd.py,sha256=17VSqkah_RldiLW8RLLI_k6KovW56Ei6a71rBJuuIxU,1095
trax/optimizers/sm3.py,sha256=Tq1DLyNKy3gO6bStVECvzrWTqjfbbtX8JmoNVthLaVQ,3807
trax/rl/__init__.py,sha256=WqCHQ1YkSJ94PXGrXv5-PGNVJ5RFkivSVgUrbHaODgc,1917
trax/rl/base_trainer.py,sha256=DfOvsAc5NDOyEH2aKwVoAAIudN9ZEjLqAFFAhE5t1QU,5205
trax/rl/base_trainer_test.py,sha256=ovdgpEbvpg_5Th0wRU4xVjeq2VwJEgxeXb7L_NyeMZs,4566
trax/rl/online_tune.py,sha256=IONVNkRhrhZPCMsOIliFOPXaEGzKXHWYh6nZ62QLOro,3953
trax/rl/online_tune_test.py,sha256=Aj4uQoueyaZ2jkzwAwiR-aT3K6SEfiNKh0XMjkxJQQM,6371
trax/rl/ppo.py,sha256=28H9IpRhNbhAlkJEqiQxLL6vMM7kzuaRvCCvrbIJexQ,38593
trax/rl/ppo_test.py,sha256=-BQ0ydwy2G7jsOExLYnGngJD_l1RPEl4eT_Q3FyojoU,27238
trax/rl/ppo_trainer.py,sha256=FNIQ5stfC0X0Bmt2v8sdnAg4BLfASZsp7HbMw7GcKCQ,32610
trax/rl/ppo_trainer_test.py,sha256=ikpQpMVnsAey_nXPRIcGjJ8xsubm9AM03xnL1r5SvcI,10087
trax/rl/serialization_utils.py,sha256=4hhuzO53OY_84c0VpoqBwyAQu6qbfz8jQYQVG6qv73E,6802
trax/rl/serialization_utils_test.py,sha256=WIWl6ghYzgC87dl4b9UKz6Y0-PJS9PAhB-HmQwfSFzs,6185
trax/rl/simple.py,sha256=RU8NFnU185iG91FwHXAgM_qE2zKheZFKgBqzdDL0DRs,8728
trax/rl/simple_test.py,sha256=k1LDYPBir97Tkl9K_RkRAKvDJ1jgfJrUo7Fmi5vIqWM,12554
trax/rl/simple_trainer.py,sha256=qJbo7TEb4707RpoUw1a_5XSspB4VuTHnkCKgwpaqwMY,12293
trax/rl/simple_trainer_test.py,sha256=wlqouJ_joW2WGINoKzDWSKZ06yWD-BdV11EJjfQ08ls,3226
trax/rl/simulated_env_problem.py,sha256=za7BQyfd-01V7XRdlJwS_0mAZYeJ40oXPmiVzxi22u4,18331
trax/rl/simulated_env_problem_test.py,sha256=uKLExYGJeKqt1ESBfPND90ZCjwMcYNnsmU9XJcZJocM,11083
trax/rl/space_serializer.py,sha256=5Vx8hpTrDwceeFrQawX3UGaSKFf1mrCPCBkluDolCY8,6983
trax/rl/space_serializer_test.py,sha256=dz7uDdu7ByDkVFEMbbX7BuegVktQfBIndsdtgNPjv3k,5903
trax/rl/trainers.py,sha256=0lii67JLNet_1oQuxErVYXo8e4KoS2YrjEcYUnPm5GY,1153
trax/rl/envs/__init__.py,sha256=amwNDn_d5mbZNx7JKgCKNE2pw0fbhEmdNHXEUAtGKVs,1229
trax/rl/envs/async_trajectory_collector.py,sha256=f7PM7UxPvgqBBs1mdjVkc00G5-gbiarxVlBvBApnIGY,5809
trax/rl/envs/async_trajectory_collector_lib.py,sha256=MJo90l3HD6SqGiLH1jHdNe-p3CoUCENYMisg7ipEqkQ,7253
trax/rl/envs/async_trajectory_collector_lib_test.py,sha256=jyXerT3lLHEyEdoBUkEy-iufxxIB328tzSD9uyFEXe8,2226
trax/rl/envs/fake_env.py,sha256=PsSLRN0u8JtIEpNF-c4MDMDKlK0-0OaX7UFFN8kVxsM,2156
trax/rl/envs/fake_env_test.py,sha256=2u5MYf-7MiSnGKkme7WZoxfpCxGuOGrzwFAs8polbzs,1942
trax/rl/envs/online_tune.py,sha256=TAkaByWh0TLtqYSWNoALFlGpbEV-V3KfiO1bB6V5I84,2086
trax/rl/envs/online_tune_env.py,sha256=cZDlnZcyZzH0sAgUvOKSt3YHCFDDkIdnmKB90oTAyhE,8474
trax/rl/envs/online_tune_env_test.py,sha256=uPfrwtKCBPom3Yrux-bGfHQ8xtfcaXl4IApBqzKdw2c,5189
trax/rl/envs/online_tune_rl_env.py,sha256=ruCtH1hY07NTT_PZuTQpYK2soKqI1muM7em4VoWbtRo,7961
trax/rl/envs/online_tune_rl_env_test.py,sha256=fiPJ9kWp6T84uG1a_dkhH1Br2q0sY1UO3QE0dskSpFE,2214
trax/rl/envs/online_tune_test.py,sha256=XZxKsVk-3LWig_bzQWlDhwHrF6zDuiKFkYBJkcs0TlU,3920
trax/supervised/__init__.py,sha256=q6WbGrD0RHSD-yTuLE2FF_qMmjyymDhYFgq9TiQKYEQ,811
trax/supervised/inputs.py,sha256=oCWgXagBxCGP2e2Hi5ThlJUqmXptQ8ooM2pP14AWccE,31562
trax/supervised/inputs_test.py,sha256=LYEWPuuN_AlHB3V8oHxcC3087YzffL0QCSC0mnoUpBE,2340
trax/supervised/trainer_lib.py,sha256=-_FQ1eongKKh4YEFdJUj51epeDp9pzMTwLu-pvIZQIg,38167
trax/supervised/trainer_lib_test.py,sha256=BPFHlb563Iqq0ujKJzz2Bz5UfQpc2XAACKFFFd6bS5E,10539
trax/tf_numpy/__init__.py,sha256=ggQWl-jKWH3ckyEmC2ZeGRdXStObvqYv0NJFBnpr4E4,597
trax/tf_numpy/extensions/__init__.py,sha256=Jq8qmEmFBXJmI1B44F6SLSmfluFdv0FVJAA53x5XVCo,894
trax/tf_numpy/extensions/extensions.py,sha256=X88Vq3i2xvdF0GKzjhDeAyyg5z3IqS0Dp3BFHtT0ihU,27825
trax/tf_numpy/extensions/extensions_test.py,sha256=LtBotIfIlwMCNXgnZRMfD9qxKC3uqejcaadlg5gvTGc,15716
trax/tf_numpy/numpy/__init__.py,sha256=DmjkYZm7RzDLfU9dDpjsNzvrNdel9GRKc-MaYx9DTEQ,1249
trax/tf_numpy/numpy/array_creation.py,sha256=6EpI1XfA4k152vaPE1v8voHnoHbTUryDijPNg_X8wZQ,20884
trax/tf_numpy/numpy/array_manipulation.py,sha256=eClC_xu5Q8aJaLC7lrPMaweSKTcNW9GQvvClFOiEVBg,1194
trax/tf_numpy/numpy/array_methods.py,sha256=I8R8ZU2NM1PDN0BWgDvjxyzaexNl-NWDnfFsZbhWej0,28282
trax/tf_numpy/numpy/arrays.py,sha256=CCBFlXNWzwEtD5YBF5yoxbUD0X4UXQIX_Eaf_DXHQZ0,11996
trax/tf_numpy/numpy/dtypes.py,sha256=rmYPuO9HZ7sOW_EumuDvq8GBIzaMNHtksaE8x7XqehM,2488
trax/tf_numpy/numpy/logic.py,sha256=NSPRaXDUeHxdUz5oHRjdS4TWDnt65ZKW4cYTzJvKK04,2706
trax/tf_numpy/numpy/math.py,sha256=m_ChCTxwJZjzLYz4h8niJXNucTORDAfqn28CLCVKPxs,9406
trax/tf_numpy/numpy/random.py,sha256=cERV8tvafX6o9-7sjsC1Fgch1bC0zUMcBj3BoFVL1Fg,1358
trax/tf_numpy/numpy/utils.py,sha256=wVROAoFbBpQMxDwddLKoVEhzAeoAvDVqg10t3HcX5S0,8764
trax-1.1.3.dist-info/AUTHORS,sha256=EkL4K1EUyZJiXwm-BZMC7gHs84ylsRyl1fhCjf-xNLI,294
trax-1.1.3.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
trax-1.1.3.dist-info/METADATA,sha256=g3wxeO5JFTY578earzPka8ljRK040TbLj0QKpaShuIQ,1459
trax-1.1.3.dist-info/WHEEL,sha256=HX-v9-noUkyUoxyZ1PMSuS7auUxDAR4VBdoYLqD0xws,110
trax-1.1.3.dist-info/top_level.txt,sha256=2rJyjFdQ3Q0lojsugL6toyALj0yhMAGuR9JkF9ElfgU,5
trax-1.1.3.dist-info/RECORD,,
