trax

View on PyPIReverse Dependencies (1)

1.4.1 trax-1.4.1-py2.py3-none-any.whl

Wheel Details

Project: trax
Version: 1.4.1
Filename: trax-1.4.1-py2.py3-none-any.whl
Download: [link]
Size: 637917
MD5: 35b71fe94663d649dbb1f94f12b5b46f
SHA256: 5884c010ceaffca5148fe702321ac8f336bc5ca9bdc88d2247a9e04f3deba281
Uploaded: 2021-10-26 20:27:35 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: trax
Version: 1.4.1
Summary: Trax
Author: Google Inc.
Author-Email: no-reply[at]google.com
Home-Page: http://github.com/google/trax
License: Apache 2.0
Keywords: tensorflow machine learning jax
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: absl-py
Requires-Dist: funcsigs
Requires-Dist: gin-config
Requires-Dist: gym
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: matplotlib
Requires-Dist: numpy
Requires-Dist: psutil
Requires-Dist: scipy
Requires-Dist: six
Requires-Dist: tensorflow-datasets
Requires-Dist: tensorflow-text
Requires-Dist: tensor2tensor; extra == "t2t"
Requires-Dist: t5 (>=0.4.0); extra == "t5"
Requires-Dist: tensorflow (>=1.15.0); extra == "tensorflow"
Requires-Dist: tensorflow-gpu (>=1.15.0); extra == "tensorflow_gpu"
Requires-Dist: attrs; extra == "tests"
Requires-Dist: jupyter; extra == "tests"
Requires-Dist: mock; extra == "tests"
Requires-Dist: parameterized; extra == "tests"
Requires-Dist: pylint; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: wrapt (==1.11.*); extra == "tests"
Provides-Extra: t2t
Provides-Extra: t5
Provides-Extra: tensorflow
Provides-Extra: tensorflow_gpu
Provides-Extra: tests
[Description omitted; length: 234 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.34.2)
Root-Is-Purelib: true
Tag: py2-none-any
Tag: py3-none-any

RECORD

Path Digest Size
trax/__init__.py sha256=CKQarlQ2DL8wS2u4nU6sGob0y33S6wVJxa65zAKO1yI 886
trax/import_test.py sha256=B7Zrnh9qa7MZDpmoiExf5CY6QvHSwQlr2eUyosne6Qs 1030
trax/jaxboard.py sha256=2vSUXX2VLMIr-nPaOrJvJ7_vLKwGopI1r7TCczXwNTE 11543
trax/predict_drop.py sha256=aG5k3QKcSjMY9X1eLxwbchG1ZSH-DJRoxhLM7a0JNL0 12241
trax/rl_trainer.py sha256=mTHsVp6BDX1BDv5tG1r4p1AqyVdKRSt1q6-MlODo8L4 3596
trax/shapes.py sha256=N9Ckj6Lvb6VLrarcYnLgO9_0p85nkNpzSiYiyyHreTc 4486
trax/shapes_test.py sha256=RAnBOwl3JuvGp2OLCRTMm-62KHPHMdXigZSbJ8btXOQ 2632
trax/test_utils.py sha256=-koRgik5EaCXoLvR9Xe9ec3ArxNmaEULzU9CdY672BQ 1093
trax/trainer.py sha256=ruk3Sv25a45TDF3Qvcz0R1gmfwA7dqn3jgBgzrSVGTg 7029
trax/trainer_flags.py sha256=kRV4f_RezOyBnVzNooCALKbVCxb9RVlYMgdol81_X1I 4009
trax/trax2keras.py sha256=EleH08M1P5A0nmvIdAB-8sjet-H3CdAPE1GrlIbi_YU 8005
trax/trax2keras_test.py sha256=_4UffaBrBV8vtapnHWoGmUmGnSJAdz62yJtL0oullic 7482
trax/data/__init__.py sha256=IPohjlWrBSHcq2mPS53lg0leU0hCv38AzM92chfP8_4 4343
trax/data/debug_data_pipeline.py sha256=Mse0JcXZyfCT7t-w8Kfj1O_x_tQlHqALVTT4bdSwLWE 1301
trax/data/inputs.py sha256=-N4GU5rHGUaYSXS3c8rXFH347yZHl6_YxoCvrEIMW08 62008
trax/data/inputs_test.py sha256=BJtVnX0EuO3wJ5Z7dlbtmv24tzD7-bLYozV2-6XVFfg 31679
trax/data/text_encoder.py sha256=x5Zes6sZj2mNuWZy1H6eXUoc2VosjXH1IDwurVlPUpw 44010
trax/data/text_encoder_build_subword.py sha256=iko4GBnGpatXBJQO2h6KTIgdx2H41-a2FT4fnJHf__A 2785
trax/data/text_encoder_test.py sha256=bly2Sbbd5qTw04E0SebQC6cTT5mAtBJsaAe6IBM_-Pc 14602
trax/data/tf_inputs.py sha256=oDNdy94Ps7cjSTlZ0AGKH9J2OLBrSW9O4cAyJJSqFVY 108758
trax/data/tf_inputs_test.py sha256=mV0eeBje6iPb6PBV0yhWMspUrh4w2U9Y2AVsNiaIJCA 31377
trax/data/tokenizer.py sha256=2HLEyL4wKOZ_WKuKGwa1zQ8QN3j4YuR6igE9ts0HGsU 5810
trax/data/tokenizer_test.py sha256=cpkF1YLjvOHR-8re4MAQ-nVd8pvQtcFcMy6FQDqBnU8 4414
trax/fastmath/__init__.py sha256=CqOVgNRWP1Fw9WIZcynSXaq5VJVAVPUmTALqVI99POk 1042
trax/fastmath/jax.py sha256=3Zerd4G7y4WIFArkyRVGVQR1O-bN52F_BHs-3_P0cWk 8153
trax/fastmath/numpy.py sha256=_lm30pTKOu6QbZIuP4eXMGzkGZ3jnHC2p-kkeeT_JS8 10008
trax/fastmath/ops.py sha256=xf2fy5ov9TDUH1iWoD8xBxYHtnRd5t-LEcGaixW7KJs 12838
trax/fastmath/ops_test.py sha256=UuQUjq4kVmb0MnGBPeIolhQyce8Ny7_9baXdHXrZdRo 4051
trax/fastmath/tf.py sha256=TjBjb4NguU34ECXrU8irNt8t6RMDb4Wj4hRG71UwaJg 6433
trax/layers/__init__.py sha256=2zXak7IJASYxXIrvALThkcfAaFfTLh0Xx7wAOICn0ds 5595
trax/layers/acceleration.py sha256=K3OqPPVwRoONXi_7UcJF3T-gm2V4JW5Ha2j7RLk3Dr8 9479
trax/layers/acceleration_test.py sha256=MJyVTra-roz62t_C2-aYaVzh9s4bKMs7aFuq-g8E2io 3737
trax/layers/activation_fns.py sha256=b667PJ_-6WKIdyIbKhDPot2GxlNaq75JHpM0wMAk5gM 8113
trax/layers/activation_fns_test.py sha256=Y9unvIcSbTkXl2-pQj1tUy5SHexKqxnXF-KZquUZeGk 1833
trax/layers/assert_shape.py sha256=htZJPojADXkBYWpy3cy3HKzGS5palumS2kt78x4KF2E 11636
trax/layers/assert_shape_test.py sha256=Na3PpvawhdLP-0NZEHpEfHGe4hd1MuB7-ZC_OLUGpH0 8102
trax/layers/attention.py sha256=Rp7F5K2PVQsLOmDIsX7xTqQkkoD37kZr5Or2_pEszOU 35671
trax/layers/attention_test.py sha256=qxhruTYNqYVKOZmTeWHuYR7u2ds_AGUx_SGcOfkjPxQ 6213
trax/layers/base.py sha256=MUGDQjHNhkP7p9FAoMQkXiB2IfuqaRV7W1w0_99Us-I 41539
trax/layers/base_test.py sha256=UmCOfdiwO89FD2dpOY-6JC_0F3UFE3aAmuyRNStj30M 7335
trax/layers/combinators.py sha256=KQkZ4S63L-ccX_oG2j1OAL02hzISdLoAewaA837EMP4 38557
trax/layers/combinators_test.py sha256=hUT127Jvc83ZUhAKNDNghTdCSE4G2U2mFhJGtQdQ4SI 26374
trax/layers/convolution.py sha256=CyRuCFIotW2c2UtNlBVRlZ_mOK_7C5QUUR9ZbWIELGQ 6807
trax/layers/convolution_test.py sha256=3T-7KeF1MLI1rd-C66bCxXVduf8-w2R7rkHrC9mlVQk 2623
trax/layers/core.py sha256=W-su_YEKgpljwpcTN2jQMmuQb0MQU_bz7C4v_SazP3A 32401
trax/layers/core_test.py sha256=q6awfv4a9-SoMfb53Ve2VzJFXegRrpWw-7bmgtGqb88 15609
trax/layers/deconvolution.py sha256=NvcVP-S_cK9_Hwl0Qq7B930K6hA1lZt3c1vIZGM1yB4 3599
trax/layers/deconvolution_test.py sha256=jDbbkE4AIIzIfJZjWe39Fi0bwZkqyynkjLfXAQnfIkA 1046
trax/layers/initializers.py sha256=pAzh66q8ITjVH3j-UCIYmbAQUSKeBE3hvUjfLhEKaDw 6819
trax/layers/initializers_test.py sha256=MXeZsYllN7euTb_CikkbtgMGGqRFOi43ZLr7tT3V3VU 3087
trax/layers/metrics.py sha256=IpN_buKdLheA4UKXXxRX9P_RHWi0cDmhTyU7mzpjKj0 25745
trax/layers/metrics_test.py sha256=wVXuut711qWPYNoNyoxTt73mAtW3N9WP3yAd-1pXGco 16991
trax/layers/normalization.py sha256=fT-RaXHeVqNstFi1F_NsaX2q3qkLASOcE98vBEir_to 7358
trax/layers/normalization_test.py sha256=RzCOPcUyVwL0QL-VaavTmaUUsRu4MVR83M-da6eCTDY 4196
trax/layers/pooling.py sha256=Rt4ClGZyelAn9th1VKcB98KgCpzbgyDSp7kV2MBsTuM 5493
trax/layers/pooling_test.py sha256=gJHaru_6pQEPmSpqqdcITpubjjLVEfuR5cL2_A8jhRs 4230
trax/layers/reversible.py sha256=dAapGGY18UzqRTU7MGf7kI4FtnTmxxzRVGjksSjcQUg 18525
trax/layers/reversible_test.py sha256=LP_L8xTjqLTMqqfvqgTMLR2lxeJ6xvZPIqqrfIGTl0w 1254
trax/layers/rnn.py sha256=fgXAsAh3ohv7e81FwL3XgxQ6uHikkTmD_GteA_eC19w 13672
trax/layers/rnn_test.py sha256=8t8zgGZF91HllavOc9HBNqq-LQDFTQCJyiSDBxGdWMw 2385
trax/layers/test_utils.py sha256=qBqZFNL_sS3WpDFcbantADA-OpsUT3X0EjTRalyo0qA 10865
trax/layers/test_utils_test.py sha256=XzSyv6CJZPYkQXHxXK0trj-o8Y-adeI7jG11aNmwivg 2924
trax/layers/research/__init__.py sha256=PnoE25dngLZMDL7wlc-3w36v-vr6KN6e-CM1muYSybA 597
trax/layers/research/efficient_attention.py sha256=8Gkyvg51LT4BvlN5RKq0l-O8NaKh_CDOcPWhYa1M3b8 160752
trax/layers/research/efficient_attention_test.py sha256=JrYIRC1MTpD_kdqMfLao1rQ9v9FwhS_89vSwHYQt0hk 17561
trax/layers/research/position_encodings.py sha256=OzDBrAvIVEbbQCa4egbXh_T0HqN_1050oOsdqYAd89c 20741
trax/layers/research/position_encodings_test.py sha256=oALvo02BmukvlFw55jHTZ1cP9oAcA4bhn3xQPGl6M48 3375
trax/layers/research/rel_attention.py sha256=fEDMKtR3IJ8JQe3CpDCOluPK8x6tuDzPG-rOTUR_tqk 19012
trax/layers/research/rel_attention_test.py sha256=YYJoMx-bLPGj9zT9MJdeEr0EA829gy9UIRsfOCZw-qw 1936
trax/layers/research/resampling.py sha256=PmysL1I6l25zEP-gFQm1D0QE_QNdVVd59rN1qo7HjYg 4228
trax/layers/research/rotary_positional_embedding.py sha256=wOyaDMJ3XTphXEUjBUHccVhfb3H8NJnlWb_JUC_FdO0 1515
trax/layers/research/rotary_positional_embedding_test.py sha256=qJgGsZzov918MTp0U2sF4dxIc_gsoZLcTolyAyjqaJM 1573
trax/layers/research/sparsity.py sha256=obe-BMlVSdhOe8VFpYboOIOMc2zMm_hvlh18obu3DmY 72883
trax/layers/research/sparsity_test.py sha256=L6cG_pEXlruhKtbCjK8__CIgLqHJI6quK6IsancRUR4 15723
trax/models/__init__.py sha256=soGltlS_VUwjdcHiLF2y_wLtu7pp8iK_ezNYEY4h10k 3787
trax/models/atari_cnn.py sha256=SMI3cdRZH-mWFPEl_1v6xJ2A2qK5PaKRM4TtlsNU5zM 3276
trax/models/atari_cnn_test.py sha256=ZxELCIx1MRC_9uojROurqGGK5b8m6CJ4jXs89oOGyec 1859
trax/models/mlp.py sha256=HKNMDWbdcJryw3jn2fbv_bfZuBqot7DK3jmAKz7jpvQ 2468
trax/models/mlp_test.py sha256=tESO5h6nEUCkLOfJBEtPE4SIFHyGBxrCd13VAdgbyrM 1086
trax/models/neural_gpu.py sha256=wJmQF2G7RPohQUqg0sN0PbPr6Ja8pYNleJudurKEYe4 2542
trax/models/neural_gpu_test.py sha256=cFeMbUTtoSwz7VMgZJ9Ba7Q_Dxp3QJfWk5_UdyIN7jg 1103
trax/models/resnet.py sha256=K5kRuT6TBv0AixWSAEjfbiBapTGNwne_mKhQzzXkK6g 5751
trax/models/resnet_test.py sha256=_13m9gJCp4YSRIUNUdylUJrdRvqNmL1WaqO28H23xZI 1353
trax/models/rl.py sha256=iBW6XNI3p0SsXKwebfwtpXwYN-zhYtu1TWDKa43I2uQ 4799
trax/models/rl_test.py sha256=ye-PznFgTEBtuOLzEtGy25ThepF5s5rhRCs_6cRW_7w 1627
trax/models/rnn.py sha256=oJCZ-ZKeHUDENl1jDMHTicneKh-8VjysG_raa93iIgw 9301
trax/models/rnn_test.py sha256=ULyFyofyNM_6sk8kp1lhcLtp1shdFclIOGgcosw4CG8 1997
trax/models/transformer.py sha256=6Tp0PxAYd7t_sqvxrRf0W9KDbR5oiH8mshDOGSDUBIo 25792
trax/models/transformer_test.py sha256=CT4cZNxQCSjbhXyWeImPBemUqKnP7LxaLD_wLJFjYTU 2426
trax/models/reformer/__init__.py sha256=PnoE25dngLZMDL7wlc-3w36v-vr6KN6e-CM1muYSybA 597
trax/models/reformer/reformer.py sha256=KD5GWwg5eIg0KwqVnj4toWzZgkHzzIVl3Aiqi0x2Nvs 25898
trax/models/reformer/reformer_e2e_test.py sha256=EyxgmJJ5iMC72QezhelMod4lM_cC7UHb5YzhUmx2flE 2533
trax/models/reformer/reformer_memory_test.py sha256=kZNaWG-H5DlMW8DlDUH_CpAxJS5CGHutgMJVnIWZKGs 3060
trax/models/reformer/reformer_test.py sha256=nZMpPc4ciSuAXfPwJZSBicQrZdyvhnAT0bCOPqmS0bI 3672
trax/models/research/__init__.py sha256=PnoE25dngLZMDL7wlc-3w36v-vr6KN6e-CM1muYSybA 597
trax/models/research/bert.py sha256=MaHMa_iLKmtXH5qjGVsU9vOfkwm18IPRtArkHDnW-Ho 10564
trax/models/research/configurable_transformer.py sha256=VDPn8t74anBYJX5W3NdGQEDt455c5x6JWwvw1AqHsgE 47745
trax/models/research/configurable_transformer_test.py sha256=ivUxhj8mRc3CR98I69GrmxgeaKX_Diry-cMY0cLudzY 6377
trax/models/research/hourglass.py sha256=L6kIO3_wmP8gjMkbWbOGuX9n9Vlpm9X-klt7Yx96BXk 12770
trax/models/research/hourglass_test.py sha256=jiUc2cryhmK8seQ_TWHBC5CTAL1SpEGhhyiK6VzpBUg 4782
trax/models/research/layerdrop_transformer.py sha256=5axPV_-zQjpsKiSZU4CeD-uPoV9Hrmeqjqj1pNhl9WU 11461
trax/models/research/layerdrop_transformer_test.py sha256=4L0RQSwdgRSjlW33dW9sW0-_XWevs4jjLC4e9XwCCmQ 3074
trax/models/research/predict_terraformer.py sha256=0x3AiAu8TB2DixuGncTj7l2gDxzE2NMptyhUYU7jIfE 6914
trax/models/research/rezero.py sha256=6z7FixPqGoR0dQ06scUbexJKmF1OcgEZCOTzkJ_PWHQ 18382
trax/models/research/rezero_test.py sha256=JM_8cyAfIH1DiRWQL0mnZ9Lhhl8O03kIejtdzZSrG0Q 2282
trax/models/research/rse.py sha256=EYOfNRND3YbZatzG6__JmygpT_KaxtQLQdlKMsxlem0 6224
trax/models/research/rse_test.py sha256=VpR-lu2ncagiaF-WLjo-25iYrmS9kecRrzr7a6qriK8 3728
trax/models/research/terraformer.py sha256=ILZXxOX7qu1AW9aLH7OieYMkE-t2pE4sTyrS3hCUrL4 20691
trax/models/research/terraformer_e2e_test.py sha256=JeyJch4kRddje_AtigEHwj9NSgCD8Nql9rbQyA5r8NE 3466
trax/models/research/terraformer_memory_test.py sha256=bNRrFGZSXTFdXFHnSBBtKT0ESwjAAnFJuSG5xyWKCCY 1081
trax/models/research/terraformer_oom_test.py sha256=hbEzq_0AkHTHCdjQzu8f7UqDhdUeKK3BTxaskoBToXc 4033
trax/models/research/terraformer_test.py sha256=CcnM4EoTj9TlrEFYYnpwQoprwKbxt4Aw4vVVPvPuLPU 8326
trax/models/research/transformer2.py sha256=IF2XlFB2WC0vThhoRnL6dliAdNIdnbznH3u-lZNJGYU 13745
trax/models/research/transformer2_test.py sha256=INX7_lby9uTRX4LxvlBobmz21ovLGRZ2NrF6JANRom8 11978
trax/optimizers/__init__.py sha256=Zlg0FRTKeC9JDwWAE-L0AngZl3vAPdLoqae7oDju6qI 1446
trax/optimizers/adafactor.py sha256=4F1z9d0gnN9pIQFqpMAKzfYuvXn0OpPiZdEpRUU9vx4 6527
trax/optimizers/adam.py sha256=V41ZdhEwBZI-MQ09OlldjmeFdctBZOpKvgIViBtJOps 3312
trax/optimizers/base.py sha256=DJR1yERm6z2363YVQlhexw376-8bkJn4EKT6_Cd5Br4 9711
trax/optimizers/momentum.py sha256=ufQAYElJkwOMahaDiOMmRJfklVpqHX82n2sdcR-GTVc 2638
trax/optimizers/optimizers_test.py sha256=tuXeJAWPmucyG0WHzBwVxp_TlnY5qBdB9AehMXTeV1o 1535
trax/optimizers/rms_prop.py sha256=YbwQoEso1F1Xn1yxroGXtkaAgofHCaEEAgqoXhK97G4 1679
trax/optimizers/sm3.py sha256=pGwbsrZnx5QW54RKteRQize4CqzDDnTby_875ZBH6Os 6601
trax/optimizers/trainer.py sha256=P_9CQ3RttV_2-ksDnkWYeu_db02aXu06smx2YmEw7kg 38625
trax/optimizers/trainer_test.py sha256=1OxfKPgOuSBTu8hf-VR-rs-yUQGwPSKdWnUnNoi_7AY 15773
trax/rl/__init__.py sha256=rEQrwjl7txhUyzQIwYy15Aj7iz7cifc2NJqphO0ylTA 1588
trax/rl/actor_critic.py sha256=GhpRer6zvRqpD2NCBDeFj3HSJfYfriZ5ZC2oGLz0T3g 47867
trax/rl/actor_critic_joint.py sha256=ilBuzLFeOjfcOUHPXkcY4T1Uec94Jj7r-Dkxxef4yYw 25868
trax/rl/actor_critic_joint_test.py sha256=TruzwcUSNutdGsgUPOo2wY1JKS00kqfSGkagMTO23l4 6079
trax/rl/actor_critic_test.py sha256=0XgsJHprvTMHBVeb7lKr5vpqF0-0AZt16KYULf6HI1w 10832
trax/rl/advantages.py sha256=ddfh4x8UXXod9fEEw8jj4gg_GWcD0o4KOx_do_487tg 6038
trax/rl/advantages_test.py sha256=L_AMx_V1Sg5eAfmZDuBIC7PN1YmNrUaEdBl9XGZnOCs 8818
trax/rl/atari_test.py sha256=uo1TSe2hFuUbbkbfZ6Ybtet5P9Z-GLofqSyT0H2lgic 951
trax/rl/distributions.py sha256=NG_Wj2WmgmacRQTPfvysc4k5HFbVpB6hF8A8Cu8h3Lw 7581
trax/rl/distributions_test.py sha256=PmQKshVBbKtZfj6Ry81z600WlPhFTypYMz8lMOA4yog 2994
trax/rl/normalization.py sha256=RcZ9_TDHZfUnhzQiokGERbC6OBUdzrCye9bCwa383zk 3796
trax/rl/normalization_test.py sha256=NipknFOlruHwRv2-dkytF10FopEKVsPorWimzxsmkOo 2333
trax/rl/policy_tasks.py sha256=Tx43F87xxlNvvXEMvwLzEKom2A2tQS8GxZBULuAUODQ 10286
trax/rl/rl_layers.py sha256=AlS90RCVplJAd8ZK6AIHBRoCZPU67yInhw5x1Sq9KQA 9212
trax/rl/serialization_utils.py sha256=6qXAMfRsAk9YQj1jPXjdg7SnZItg3Z0eViPgyBAHsTQ 16076
trax/rl/serialization_utils_test.py sha256=poKDgYgidwTveXs6e8YSpWNysBFfyrZU9_TCMNDJfmw 10269
trax/rl/space_serializer.py sha256=ugdSXIfiGyI5EfRD2LLrrpsZGg9A1SN9_5bMDIaHKNI 6874
trax/rl/space_serializer_test.py sha256=2nuYB4OKlWeb_oNPurnz2IfwlXdmGPo0Lm9anJ88Wag 5783
trax/rl/task.py sha256=rJ5y_vXwHepuA-TtpR6JA_cDLypE_4oF6s4KuFfQY0E 34229
trax/rl/task_test.py sha256=ezOELQKr6PxUG2N1eayeVZeb8V1JrQadcYc7DfLppvA 14553
trax/rl/training.py sha256=PDuCma0DLoFkI492MpORuRwOdVFiwSg7l_kFkLKsTQw 46722
trax/rl/training_test.py sha256=rB1Vl2fqnDkArgslv5oEhEosAarAFpEAp_-bh9bDC_w 5938
trax/rl/value_tasks.py sha256=0BbOXTRqfDrrATx7xdOtkbJD3HF2Rm1rcyoXju1hjSo 8330
trax/rl/value_tasks_test.py sha256=3Xi_obYu_OdQaxNSYybUV400rl0rez1dwSik4tZOocQ 6700
trax/rl/envs/__init__.py sha256=zhCF6TkmHA1j_wQuSN0E38E9kQwJaWGYlkiwTU0EQ9U 927
trax/rl/envs/data_envs.py sha256=tpHeUys6-RVrTzo6nhKvgnSX__aRHHy02KnwThBOj3s 6306
trax/rl/envs/data_envs_test.py sha256=UkOm71mTmi8IBiAlTYk6WmmEuMOS9P-UTSVtvSlLfJE 6302
trax/supervised/__init__.py sha256=jVULxyFlfH5yXP_2yj8Kl22CFbEe28ARpRa20cBBDtw 1021
trax/supervised/callbacks.py sha256=XXjkJIsdl6vWMznv86tHgmuSiMh-zY5Tdl1YSUv9JhA 8220
trax/supervised/callbacks_test.py sha256=K4_znOXyWmm1Y0nJ3V3hFF_pdOu7xZ87L6t-EyL9KOw 6870
trax/supervised/decoding.py sha256=7iuqmUC0pERtW15uE_fnItkAzU1mO-_2itSU2E0YTVA 12411
trax/supervised/decoding_test.py sha256=scRM_1BFrW-nXIXO-U4pYxHA7tq04BjULxdYYqaJW54 18187
trax/supervised/decoding_timing_test.py sha256=67ptLbr9r37YRKZieGaTqOmNRa1OOtVs6BbhPGCdEdo 16167
trax/supervised/history.py sha256=1XeAmwCeigcmRKVFOxYb1HaluhDOt8mdN-ZP450QhOc 2595
trax/supervised/history_test.py sha256=HrDqT27I1DjAJ7Vd_dHk4N8eoEit9NZrT6oJMW-Ab30 1946
trax/supervised/lr_schedules.py sha256=yqieu7liMiQWD8xuGeMR8dNQdp1grdeUwvTYy-HazD4 9104
trax/supervised/lr_schedules_test.py sha256=csoKV2cQm_n47MDQWLdhZSX3vi6IFNS0T31xjVxIYNU 3304
trax/supervised/mnist_test.py sha256=f0uGk252THIbnkAc9zAteyGN0u_uUopz4FvOQyUYXGw 5326
trax/supervised/pretrain_finetune.py sha256=ntgxniQoRDWQ5tAGwIbXyyVNuhBw096qdS9vfvARHLo 6243
trax/supervised/trainer_lib.py sha256=uOynI5FsbjgrYIwMLFj48wNHY2kYl9ygehBpoTSBCuQ 37862
trax/supervised/trainer_lib_test.py sha256=pGVpGsvfvPZZYaPmM3L6Gf_qld3vgr5pxrDN_Icrq-0 17411
trax/supervised/training.py sha256=AWspNlyQsCfTthYVc3Fl5qYKzb25wwnLbGgfDswlXE0 55553
trax/supervised/training_test.py sha256=JDxYMtgG1GdZD4IyPfEBv8J41E3g69yvlbr_cbxGFr8 27037
trax/tf_numpy/__init__.py sha256=PnoE25dngLZMDL7wlc-3w36v-vr6KN6e-CM1muYSybA 597
trax/tf_numpy/public_symbol_test.py sha256=iZm3YKF7RT8PDqV8mfAeURo-eeW7IFgbZTS3k2pkTQ4 1130
trax/tf_numpy/extensions/__init__.py sha256=Yjj1azM8MvNz0kfeoxAqTWNYq4B5xBFaReLhk5reOM4 784
trax/tf_numpy/extensions/extensions.py sha256=5HBzk7QrmPvzR8DlIHjzJqMahon4pUuGUP-XM5gw-wM 71765
trax/tf_numpy/extensions/extensions_test.py sha256=FnItyX3xpSxNRB9fOqDcniSv-LX4jTh5eqUBjewaauM 34294
trax/tf_numpy/numpy/__init__.py sha256=JnkF6CUtxwMcntk774RJE0X8zNL78EX0ztTXxtfOCqk 2613
trax/tf_numpy/numpy_impl/__init__.py sha256=hb9_Dg4nazFytkW-O0oitD3LqbvVPXEJeGAuB7hD42o 626
trax/tf_numpy/numpy_impl/array_ops.py sha256=kPhEzUqCkpV0mAJnsDAGzVh7pb25EYpuj9IWhPgXClM 48444
trax/tf_numpy/numpy_impl/arrays.py sha256=DeEQ_y9MdHCQGZb3PisObKyINOVJdNVyZTHtunKbhIk 8704
trax/tf_numpy/numpy_impl/dtypes.py sha256=HRNo09KOAocjB7w4UTXez6ixgqbgNyFqmBm0Sr2X1ns 2342
trax/tf_numpy/numpy_impl/math_ops.py sha256=a86V5-vm3BXeup16WkZ0SwKRwRSOoDNJtxRPmF1p4RM 31568
trax/tf_numpy/numpy_impl/random.py sha256=mYZcveFu-yIaUW8VYLuWzYxMofbuJ3ZU_3PJfcE5YMk 1363
trax/tf_numpy/numpy_impl/utils.py sha256=OJbGa_lByAaDY4OtrPpsAY1KsTUcoKc-g6sjXVc23d0 10976
trax-1.4.1.dist-info/AUTHORS sha256=EkL4K1EUyZJiXwm-BZMC7gHs84ylsRyl1fhCjf-xNLI 294
trax-1.4.1.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
trax-1.4.1.dist-info/METADATA sha256=3bkb2wtfDkln1--Wg6e5F2DvbygnPRVQa5TN5UVddok 1692
trax-1.4.1.dist-info/WHEEL sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY 110
trax-1.4.1.dist-info/top_level.txt sha256=2rJyjFdQ3Q0lojsugL6toyALj0yhMAGuR9JkF9ElfgU 5
trax-1.4.1.dist-info/RECORD

top_level.txt

trax