Wheelodex » Projects » learned-optimization

learned-optimization

View on PyPI — Reverse Dependencies (0)

0.0.1 learned_optimization-0.0.1-py3-none-any.whl

Wheel Details

Project: learned-optimization
Version: 0.0.1
Filename: learned_optimization-0.0.1-py3-none-any.whl
Download: [link]
Size: 244488
MD5: 8c920ee520d3ea66f96245e50f79d1d2
SHA256: d274b71dcbccb130a1e124ea28cfec82edebc43f11240f564372ddb116194370
Uploaded: 2022-03-18 18:46:37 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: learned-optimization
Version: 0.0.1
Summary: Train learned optimizers in Jax.
Author: learned_optimization team
Author-Email: lmetz[at]google.com
Home-Page: https://github.com/google/learned_optimization
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Requires-Python: >=3.7
Requires-Dist: absl-py (==0.12.0)
Requires-Dist: numpy (>=1.18)
Requires-Dist: jax (>=0.2.6)
Requires-Dist: jaxlib (>=0.1.68)
Requires-Dist: nose
Requires-Dist: dm-launchpad-nightly (==0.3.0.dev20211105)
Requires-Dist: tqdm (>=4.62.3)
Requires-Dist: flax (==0.3.3)
Requires-Dist: dm-haiku (==0.0.5)
Requires-Dist: optax (>=0.0.9)
Requires-Dist: tensorflow (>=2.7.0)
Requires-Dist: tensorflow-datasets (>=4.4.0)
Requires-Dist: tensorflow-metadata (==1.5.0)
Requires-Dist: tensorboard (>=2.7.0)
Requires-Dist: gin-config (>=0.5.0)
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
learned_optimization/__init__.py sha256=Bj-scXC-WGdK5eq2KpCq0e3Vn7XycZPIMD567nYdpRE 623
learned_optimization/checkpoints.py sha256=Don299DrFYoGGTGdxo3tMpdko3H96PoDhIsp0xGhYmg 7162
learned_optimization/distributed.py sha256=o9qTChQRR0jFDa-kuXXX4AOnGmkhR12hwBDSmNkEFrM 8929
learned_optimization/distributed_test.py sha256=EhKeuzFH6HA8tuPouQQEwmGVbyY5twEMuDh-n90WAYI 3003
learned_optimization/eval_training.py sha256=WyIWpqdXd9q2gQYClzMV2wJN1wo_KdLBRMzLiIRVyUM 17014
learned_optimization/eval_training_test.py sha256=_GmDkgEdKrtyqQEpGtbMyrgRTXyWYnXiR9LTq6Sl8Q0 2652
learned_optimization/filesystem.py sha256=QUY-1lxtfrJtARzl92OSFwlAw7TQ2hTEJCAbHxw2PAw 1484
learned_optimization/outer_train.py sha256=NvG1PYWXNDk912uInrjaUUqxtliEiRZoyT_56zZoPzw 39447
learned_optimization/outer_train_test.py sha256=drO9MEceVrz6NAe5y9M_iN-AK9kD6SBKgYva7KjXUJY 5965
learned_optimization/profile.py sha256=QxtBMqlyyb1s1uxxZBjYLs9qjE7Tx-v-QzG8Qn8NkKM 1613
learned_optimization/setup_experiment.py sha256=XXIhWkQDIGHrUgr9cMGZBis1x2HizylIm8mJNnROxRQ 3372
learned_optimization/summary.py sha256=W_J14PcnjNRDLHTYERU9s6yfGecCFPYxJNfvugZriL0 11770
learned_optimization/summary_test.py sha256=polMAc19iPwJzP6WoyXh0LLSrE7aEUVsmQ5WuzrFeKg 2289
learned_optimization/training.py sha256=QUIkjGjN3UHyaSbP62RxQzc__IM2o6b9IoHuW_QCWJU 3257
learned_optimization/tree_utils.py sha256=4J4ZVBuIuKBmTJLdKAVHUJe24Puo3HjU0zw9FFiFI0w 2980
learned_optimization/baselines/__init__.py sha256=DWNFWO6IWY9UIz-SNYNuUlzsLoWQWW_u5ns9zBFy9lY 615
learned_optimization/baselines/hparam_sets.py sha256=9kvDRrqJddEv7pC8XwOLZWcDAtov5mVADSGFOifhwqQ 3045
learned_optimization/baselines/run_archive.py sha256=BhJ_-p-dihCHSAQXVEDzQoxC8A5_acy4IWHbjkO6ZCQ 4069
learned_optimization/baselines/run_trainer.py sha256=tyNf2qSg0ptG-rGXqiLcNqxFQs2-W6ECxRyGL6gPrsw 5262
learned_optimization/baselines/run_trainer_test.py sha256=cVDYDcnxXQgFseh2ecLoBQ9fcSgON9J5r4Ltt3pw9uQ 2838
learned_optimization/baselines/utils.py sha256=c7b-RN322C37waGvdHzl4wEF4wjlFDt5X5Ru9N4zwec 6575
learned_optimization/continuous_eval/__init__.py sha256=3YNbqOIPU-3D7-z3xhFVJvQAp0EHSwaAgXhWBR_HaiI 628
learned_optimization/continuous_eval/run_eval_chief.py sha256=lzuR-H9C_KJligahT59TU62_5HsUUL_XOf4epz6XUiA 17294
learned_optimization/continuous_eval/run_eval_chief_test.py sha256=RZlsvYsnRfHlhn7M2oSnX1FEim5qh6braTxiV9k75nA 6583
learned_optimization/continuous_eval/run_eval_worker.py sha256=kjj1GIsQJVtLQQ4RpaRWPt6wdZWXtQIXtVLYeJe_PN4 8782
learned_optimization/continuous_eval/run_eval_worker_test.py sha256=yCGfpWqHecVZgqDBApSD9TVhayLWpt0uU_UaX4ihAyI 3007
learned_optimization/continuous_eval/task_group_server.py sha256=j1KmJE0hVU7MMDnFtQRigLcpoeaQpeVDfXQKac-QmrM 9810
learned_optimization/continuous_eval/task_group_server_test.py sha256=GTL-t1plHvgG-yV14MQAnKr8f87Gp77-JOL5egkA-Gc 1759
learned_optimization/learned_optimizers/__init__.py sha256=MkRY7sz12G4vx4Xuu5FbmncfF_1TMoQQk9GuScsJgBc 643
learned_optimization/learned_optimizers/adafac_mlp_lopt.py sha256=rFBHIEm0LhAHhCL9hzVneBZkZi3njI4uiT9RFqzQoNQ 16000
learned_optimization/learned_optimizers/adafac_mlp_lopt_test.py sha256=EiBm3D_x4_8E8Hz1rncvJDvjOjkPMCIFg4_P7t5jTIM 1784
learned_optimization/learned_optimizers/base.py sha256=a6sTK566WyLDOgAnXHRuAOxaNOC7ugaMGnojzqTz9_8 5725
learned_optimization/learned_optimizers/base_test.py sha256=8G7WQsULhZTs5_vjJgLgnDy2-R5IEw2mlyd_a3ztAiQ 1184
learned_optimization/learned_optimizers/common.py sha256=5mVcznnjNrz2fVX7mgLVm6hYWEifaVzEnOd6KTIy6oA 7056
learned_optimization/learned_optimizers/mlp_lopt.py sha256=lQ89A6ee6fm2gm69a7u8tgY8TYOfr7TP4OctEYBOt08 7075
learned_optimization/learned_optimizers/mlp_lopt_test.py sha256=CDQiKUQd9_WfGy-_vYxfsYcnkplv8h9l9BE1mGCE3Lg 972
learned_optimization/learned_optimizers/nn_adam.py sha256=YzH8-Hz75ajzswbAPyqxEl2UuUxAIAYh7zXkR4gPPuk 16943
learned_optimization/learned_optimizers/nn_adam_test.py sha256=R_wEoUU_mWM26Ji_pClIvmt-YZElZkp4JXz9o69FP6U 966
learned_optimization/learned_optimizers/rnn_mlp_lopt.py sha256=Mefavxzy1MKmRVgo3C9ZMsS1MJAB1qYaQw8kZ9vdObc 23360
learned_optimization/learned_optimizers/rnn_mlp_lopt_test.py sha256=amVaJeMWgtudnEMIvJZmTsqeM8RcCHbkdtDPCgaLhqA 994
learned_optimization/learned_optimizers/test_utils.py sha256=PyKjSa0lD1kUHsh2gtCpRtD1QPy-HSPpx_dEWR1fOOg 1002
learned_optimization/optimizers/__init__.py sha256=fR72-l69KBqh_H_4JPMUuKydkyxG1kmIsB7FOkzi6Pg 635
learned_optimization/optimizers/base.py sha256=JX2A2csvGNj8W5Wt1lt-5-mCERUnNOvUFeajJXJRRNo 7924
learned_optimization/optimizers/base_test.py sha256=lMq4K-oC9ZcpiByYbPrbRKGAWNcTPFmohC_TS7MvxEg 1387
learned_optimization/optimizers/test_utils.py sha256=sT1zV6Jx_IY--FwchGai24Lm7k0z5Ixtrrv3GW1AtHY 3222
learned_optimization/outer_trainers/__init__.py sha256=iDBiiMqqfewPmTf0BLLBxkJW0sPDLe6xWsBEstC5dgs 639
learned_optimization/outer_trainers/common.py sha256=1TYvM8p8KzZ61ydss7j3EhM2GtZOMkcOQjQMmMYfPsw 12103
learned_optimization/outer_trainers/full_es.py sha256=t-UqFcKsTLsgTDT9Zbh0AJdv2Ms77RdgxQ5u3oKQDQE 24248
learned_optimization/outer_trainers/full_es_test.py sha256=H4TWGpq4iXXdR6UACi_FNT37DHgdywzSjj_t-_4lino 3804
learned_optimization/outer_trainers/gradient_learner.py sha256=LTzr_5M2yyG8kqiI0_jgUfUcYJEn5KJGJZU1S7hBCaA 17720
learned_optimization/outer_trainers/gradient_learner_test.py sha256=Agdg_wNPqBV66XNFJ4qOsARV3If7XfWx3p9gmKjylSk 6060
learned_optimization/outer_trainers/test_utils.py sha256=Lmwj0JfN4RtKCDQq4XVYT9Cu1PbtiLpR0HVzR3-X-RE 1097
learned_optimization/outer_trainers/truncated_es.py sha256=tL4tjm0pD_md2jkZ9swJJz2H7_vNCOFjH3-cUmDsAU0 10335
learned_optimization/outer_trainers/truncated_es_test.py sha256=4XxCHO5gbED6SO5vXUfgiIEU64a65VJSx6-4sjsZIug 2061
learned_optimization/outer_trainers/truncated_grad.py sha256=rP8mnSwIYh5d05Ho7fh18fsGCpmXa2S_RNSoYnXV_f8 7401
learned_optimization/outer_trainers/truncated_grad_test.py sha256=5x3Q79edpfGJtutlPHjCjCiP4MlQ5-xdxgxDphz-H8E 2079
learned_optimization/outer_trainers/truncated_pes.py sha256=Dfhs49PowUDptggHBOoIC88kkF-WwCT3JAdKJZ5hb98 18983
learned_optimization/outer_trainers/truncated_pes_pmap_test.py sha256=DrOXZR5wm3xx2jv2ZMlipfweSKEdF_caGmkZoPckA0g 1960
learned_optimization/outer_trainers/truncated_pes_test.py sha256=cEduoqxyekS9zJajp7O_PK0p1PDSZx43ghCnCuyrdkw 2455
learned_optimization/outer_trainers/truncation_schedule.py sha256=5jZ6OryM2nDpvPjkK7P4caTjQvN8E1env-RsoFW76YM 2607
learned_optimization/outer_trainers/truncation_schedule_test.py sha256=tVN2k3YQKD68IRvkKpHO4z9JWFjPJqm0jCufTE90zAE 1673
learned_optimization/population/__init__.py sha256=LG5N0XWu9TyK4ykTzBI5gwLPr5Gcuhp45_VIHfGRtvo 631
learned_optimization/population/population.py sha256=qMyWZXQvwsDdgn7mlgoRXKkGgUjbo1OL9F0_OHUpneY 10872
learned_optimization/population/population_test.py sha256=9MTeEQ-k9e-lJBqyO9JrbdW38LEbpTjAVeawJDARgYI 4345
learned_optimization/population/examples/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
learned_optimization/population/examples/synthetic/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
learned_optimization/population/examples/synthetic/train.py sha256=PXIB6FpBwhn_52cYcqDKahfw6vVzxZwcXFvuAA-J8QA 2558
learned_optimization/population/examples/synthetic/train_test.py sha256=fYTdcZmiYom0Lw3N3W1QyWam6HkIcrPhr5PtOFuZMxo 849
learned_optimization/population/mutators/__init__.py sha256=cIy5D7Fv_CAdRsdJxekovWw5vApjiR-bWvNis0meKzQ 614
learned_optimization/population/mutators/fixed_schedule.py sha256=vKCrPTS89cBm6uMGpEH2ktrc7ReU4UWIEpaNi_Ar_BE 3316
learned_optimization/population/mutators/fixed_schedule_test.py sha256=0rX-MwAvedUe1uYhXZ4x0R6ayg5b4ipFTRXwQgJPHj0 1733
learned_optimization/population/mutators/single_worker_explore.py sha256=0ykveICHcoAngb5hyiDey3_dtM9d46L24tMnOgwfNoo 7950
learned_optimization/population/mutators/single_worker_explore_test.py sha256=N0C_qH9xtNHjA8PV2H6ue0tuRUEZqdL66m8igPR5Pqg 1666
learned_optimization/population/mutators/winner_take_all_genetic.py sha256=Z_486Aa7fx0aAwv-LkyEo0-H2pm3aPMeKENC36Nf-bk 4610
learned_optimization/population/mutators/winner_take_all_genetic_test.py sha256=whR8PY9t_78Ft2RgqOUrNj0Xun_Ap1rXKn9xTSBv-so 1714
learned_optimization/tasks/__init__.py sha256=rEGSDZVaWnCMWlWG8VFg71rA8ZsbkmUt4bja0-RG6Vg 630
learned_optimization/tasks/base.py sha256=CqRpfwpPqOb8DHqTcTP6P2BU0Hcc6LDYUflIP13OcQ4 6283
learned_optimization/tasks/es_wrapper.py sha256=X_76jLFznlywdPKFsAW4Pj_0tMxdeEDZvatkwZkxJuU 8772
learned_optimization/tasks/es_wrapper_test.py sha256=I5EuuWx59MoYUNQCo8lIIquiIhws_URKWcPsilSGGic 2848
learned_optimization/tasks/generative_model_utils.py sha256=0cXmHh29Wye-Hl33VqCPhAjHuFqyyP-bm2dQAyCL18g 4181
learned_optimization/tasks/quadratics.py sha256=QJ0rkVJSu4fbnG1ClVVxKvXhvoxxkTQiHeKOOmxu8H8 6238
learned_optimization/tasks/quadratics_test.py sha256=y0e1hp9oK6yVlR5zbZXnxs03NXss0ZASTKR8Q9jz598 1439
learned_optimization/tasks/resnet.py sha256=Tcls0fk4aDDxrT91CulIkHRFvdKWWSw5zY9hOToP0nc 12084
learned_optimization/tasks/rnn.py sha256=4gZW4NC9_HOtfia8ejC49IpgwOcAM5Bp1mZxRIbHPrg 1827
learned_optimization/tasks/task_augmentation.py sha256=-Bq40YQqYpqUQpmuQzKfmDw-pC1-ximX8PD6WTUD-HE 8843
learned_optimization/tasks/task_augmentation_test.py sha256=mTtHy666j7cYAXZlmTWwG89nDpgmU1ZD2JFuXZfHi1w 4880
learned_optimization/tasks/test_utils.py sha256=3weenVtj8wD9ahYje7pZ2BFs5DuKYkDZP5zkSwdbIjM 2505
learned_optimization/tasks/transformer.py sha256=7lY1ZKprtw1iY_VxL7wbKcBQbklBw9xscDj8hz80R0A 6226
learned_optimization/tasks/datasets/__init__.py sha256=JDY254ndmtME0aizLU_uYBSeoPoCRT3QrVtHp0DFL7w 633
learned_optimization/tasks/datasets/base.py sha256=tevlQqJN4muM8jmrZT8q0twxq8ryhISG2gREU5Ke7yg 15164
learned_optimization/tasks/datasets/image.py sha256=UYq3Nm1OUx8CiuqccnHRtvneCJf27kPnDZMIzyIFWag 4183
learned_optimization/tasks/datasets/image_test.py sha256=_5-_tzP6dMAgxWsBNby6TqfDm2X8rBe-gzaAFIUT3Aw 2800
learned_optimization/tasks/datasets/language.py sha256=1fp9EFJ3cwlaQbEGa3h2_Bl_SdF82X3sIT4HNV-VWi8 4244
learned_optimization/tasks/datasets/language_test.py sha256=e8_8o4rJMDwz-96ljWWTCS6_iCXORaFBVlRHu7ViWZI 1911
learned_optimization/tasks/fixed/__init__.py sha256=2TdA6wEJX8_SWZy_RcnpoNTRaiMPqFMmn_knLHVzpWM 617
learned_optimization/tasks/fixed/conv.py sha256=rHuWXNIREizn0vtaWOyopqen_DvaDRbvwx9bV1A7qWA 5982
learned_optimization/tasks/fixed/conv_test.py sha256=Btt_SXNSOQ_lh1Xww4GlNGr6D1Rx3UrK5CTieeAtl_4 1315
learned_optimization/tasks/fixed/es_wrapped.py sha256=hLNG2hsKPRYB4nJocNFQ3yWd7nMLH1KW8CKX9cBehLQ 1476
learned_optimization/tasks/fixed/es_wrapped_test.py sha256=2zBrjWmyswSD4NBZi0ndee6Sacf6ULN3wqJ_pSIYroE 1312
learned_optimization/tasks/fixed/image_mlp.py sha256=80g1AqABi_Ly6q-8ZNm08pT4EyM3nakxyaKTFpf8CJk 8433
learned_optimization/tasks/fixed/image_mlp_ae.py sha256=Tq0FhpT4QZiV1TvdIIdPGEzgudly_1A_0NeEGDnkZ4Q 3839
learned_optimization/tasks/fixed/image_mlp_ae_test.py sha256=FJxvJvR3v95ZziOETgn3NtFX7bPPLjLEH9yKDVhSGMk 1348
learned_optimization/tasks/fixed/image_mlp_test.py sha256=E1xHJT4V76U1FDvHTh7C8Ujb3dlaJ-NVRB3xiCi7Z-Q 1363
learned_optimization/tasks/fixed/resnet.py sha256=khcXqUA9Wxajkio067z7zYt7DkwRRgd0PyuW_cWUbe8 5931
learned_optimization/tasks/fixed/resnet_test.py sha256=tHBV0OtrJAG1E1HJUBFeXf7lKuHOJ5wQ3Uae20fCXgw 1195
learned_optimization/tasks/fixed/rnn_lm.py sha256=M--KnEvdyQMEyDAEpIHZN-6pyDtGmBmD4tZDdJ6wBQA 5897
learned_optimization/tasks/fixed/rnn_lm_test.py sha256=pvGOqqUAoJDVJ-ZlZCD5h5-c8lnGpx1S9--9C7fghUs 1762
learned_optimization/tasks/fixed/transformer_lm.py sha256=nzRm9QZnmThOtxrAifO2v5LHS9PHsfBRL-Z8kfLniG4 4608
learned_optimization/tasks/fixed/transformer_lm_test.py sha256=TKZOCvIjfhG_v-BUg_BMJMWnpofXY7bFPoFwTBbMwDA 1346
learned_optimization/tasks/parametric/__init__.py sha256=rk9mXTT4pVFzBu9stF3jBSp3I7T_TKmZCC4ReTHGL94 622
learned_optimization/tasks/parametric/cfgobject.py sha256=hOP1svhEmHuz-gyhqSWI6Aidun9nfebYV0UeCR8VROg 9000
learned_optimization/tasks/parametric/cfgobject_test.py sha256=U_NsJXvyeZ3N2tPLpHFIG9TaN7_9EBSChA4jy2mS810 3148
learned_optimization/tasks/parametric/image_conv.py sha256=fNx2QNF0qkiam_9un1ViRMH4eO58Qqm31XcXbs_BKUc 4830
learned_optimization/tasks/parametric/image_conv_test.py sha256=voRVYmMYU9vONz7ENSXruyTQUDHNJNfIUZrjXwuB510 1524
learned_optimization/tasks/parametric/image_mlp.py sha256=tLHfx1mLzfSSLtYR4814D7QQ7L_x0pn7Z_V3qDHlvx8 4658
learned_optimization/tasks/parametric/image_mlp_ae.py sha256=Eyy6nL1hTILp76r_1X0ukx8mQcEklOVZu5BGEQBop9o 5247
learned_optimization/tasks/parametric/image_mlp_ae_test.py sha256=5j-srfNgJTzei_3vtf1A00cEiOyyQRI-ulStgOFqUt4 1634
learned_optimization/tasks/parametric/image_mlp_test.py sha256=-liXEnuUacYyjOi7m3bcY0U64LCYDMwI2tmDkzcuT3Q 1384
learned_optimization/tasks/parametric/image_mlp_vae.py sha256=q9bY8oQb9xvN7YomvdoOZz1y5D9gI1MD3eUjcqMrFUU 6150
learned_optimization/tasks/parametric/image_mlp_vae_test.py sha256=W1z6cgj7wbpHH6wC1ZDyZonHgI_DUXnVltTzBUpMKyA 1705
learned_optimization/tasks/parametric/image_resnet.py sha256=iOtGTyM01Awf_TACY6eSDAxNOBS_F6pygu9Bdsk_Y8I 5628
learned_optimization/tasks/parametric/image_resnet_test.py sha256=OxJXwPnjnpgcS7LBo4zuawPuzBBxX5xpx1bNN1zWxP8 1851
learned_optimization/tasks/parametric/lm_rnn.py sha256=KZdBO6kyRnOVE4lPSzvBDVo8HoeJDP_5YoasgHUaz8c 6705
learned_optimization/tasks/parametric/lm_rnn_test.py sha256=V6v9wECz-6DH05NHiQsvCxsQN0LYS7UiQaQ61bOsM3M 1662
learned_optimization/tasks/parametric/lm_transformer.py sha256=0NBorEhyP7p6HejExM-Ytkx1dUKPqPOYUzaxgHFwPEE 5126
learned_optimization/tasks/parametric/lm_transformer_test.py sha256=JFnkx1PxUJ7UeXidu1Cjh8zk6sNoaVuynGoA9UA2LnY 1712
learned_optimization/tasks/parametric/parametric_utils.py sha256=bdP_hIJVFe9lyh15jStK3WJn1xxjHzre8Uc5l9W5uaM 7940
learned_optimization/tasks/parametric/parametric_utils_test.py sha256=fsaqMMV3aEkrmKEcXKOdt5Zvi1dHn7HEyOESDa5GXZ0 2305
learned_optimization-0.0.1.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
learned_optimization-0.0.1.dist-info/METADATA sha256=N7hKZx8vNf2BzeMp84355h3Jyu5Hu014Mu4SGDfXv94 1001
learned_optimization-0.0.1.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
learned_optimization-0.0.1.dist-info/top_level.txt sha256=tvUk2rVafdCPo64AVKD4dPbEoKwXIBRhYStoQ0X5rkU 21
learned_optimization-0.0.1.dist-info/RECORD — —

top_level.txt

learned_optimization