praxis

View on PyPIReverse Dependencies (1)

1.4.0 praxis-1.4.0-py3-none-any.whl

Wheel Details

Project: praxis
Version: 1.4.0
Filename: praxis-1.4.0-py3-none-any.whl
Download: [link]
Size: 772480
MD5: 913c54ada03ebd87ded30cf227e60763
SHA256: 787bbc11d65e4a60a2240cc7ae78e56b81cbd99f88fd27da474c730fc8b3b2f3
Uploaded: 2024-04-09 22:29:07 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: praxis
Version: 1.4.0
Summary: Functionalities such as a layers for building neural networks in Jax.
Author: PAX team
Author-Email: pax-dev[at]google.com
Home-Page: https://github.com/google/praxis
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10
Requires-Dist: absl-py (==1.4.0)
Requires-Dist: chex (>=0.1.85)
Requires-Dist: clu (==0.0.11)
Requires-Dist: einops (==0.7.0)
Requires-Dist: etils (==1.7.0)
Requires-Dist: fiddle (==0.3.0)
Requires-Dist: flax (==0.8.2)
Requires-Dist: jax-bitempered-loss (==0.0.2)
Requires-Dist: jax (==0.4.26)
Requires-Dist: jaxtyping (==0.2.28)
Requires-Dist: lingvo (==0.12.7)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: opt-einsum (==3.3.0)
Requires-Dist: optax-shampoo (==0.0.6)
Requires-Dist: optax (==0.2.2)
Requires-Dist: sentencepiece (==0.1.99)
Requires-Dist: tensorflow-datasets (==4.8.3)
Requires-Dist: tensorflow-metadata (==1.12.0)
Requires-Dist: tensorflow-text (~=2.9.0)
Requires-Dist: tensorflow (~=2.9.2)
Requires-Dist: tfds-nightly (==4.8.3.dev202303280045)
Requires-Dist: typeguard (==2.13.3)
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
praxis/__init__.py sha256=yRyT7qjhz2-cckhBdD_4WvjWbUZOBSr7gEo_VbuRdLE 596
praxis/asserts.py sha256=AI1pqW-04VQwiCp76Woqhi-6j6bJd6iXwDm7EBX3sxg 22000
praxis/asserts_test.py sha256=x50QaEq-_QO9e1xmJmVixFDPiEJOTP6r860Dw3L5Tlg 11563
praxis/base_hyperparams.py sha256=5JeMncPN253ZxSFFVU5ajrOHXKd5S5nmP76rXh--n_8 44079
praxis/base_hyperparams_test.py sha256=SPu1cF00kDoDeNNTaVaSS1h5oMnS0e1AAQBOLIRIwYU 27429
praxis/base_input.py sha256=nVfefZxJ6e7oyuF4s5TZ6Sdn8HuV74h4W9JuUQqEiqI 44661
praxis/base_input_test.py sha256=O3IpGO6qwcQEIsMTDK9Qm8wd1n5zieL5SleSmigbCBM 34620
praxis/base_layer.py sha256=SCwYeGMwk3pd6ZbCGnB2Bv7biC40CD4_MgT6EpolFaQ 100935
praxis/base_layer_test.py sha256=3B-b1KEai5ENq9TduaWmTiOpzcGvaIu3x0dr3JkkH-c 31284
praxis/base_model.py sha256=5zcrGSbapXMHcPRwrJvlvIPjFhNlWtNHsMSpeJe1Zrg 5773
praxis/beam_search.py sha256=rZX92tXklUUIrd5X2eheDOHWwQHwTtcCV8k7iJLmkDQ 19218
praxis/beam_search_test.py sha256=dSU2lBPOJg-qRjc1z5uAcyLVurojWXryR0x_66XG2II 15704
praxis/decoder_hparams.py sha256=y5a1Zxg5oSIMbj0NlUii6fdzem7M3o_qHcLQE7xpCro 6540
praxis/decoder_utils.py sha256=oWeC5rHHbcKqGIaOHaWglIFF3KYsfRZADD5cncKD7Jw 25777
praxis/decoder_utils_test.py sha256=dY9EIcP9BIl9TUeZx5U_Tk7FKdjxx5mts8b_Usnf4HE 12772
praxis/fiddle_tags.py sha256=9HTB1o8TUQDrnCt_STgirwRtOIMFQu4cdqcWL9iyggw 1475
praxis/flat_beam_search.py sha256=DeZE7oY75bqceh6fA-1F1RQYzmxG3oQhJSHUYbDVy8Q 11591
praxis/flat_beam_search_test.py sha256=voe4EReciYT6Z7xz5NPG1wn55QBnrlMwUlwPfvEmTHM 4407
praxis/flax_utils.py sha256=J34OGy9qqQ0gxA4Mhoyubi3D9BbS_wsgXIwFxtKhhZg 5602
praxis/gshard_utils.py sha256=i_KqV37JBob2Q2dGKE7wuaVPew3cEec9s02eqzAojxg 35272
praxis/lazy_loader.py sha256=fkRaTuSkRMavbyraF8yOpO-f4B2YkgThsFWp0vMxQTE 2600
praxis/lingvo_lib.py sha256=S7KDhY_hVRwtlJP5R0MLwKF1qa__VgFJXqZlk3EJmNk 1555
praxis/metric_utils.py sha256=XHfvtuSGeRBkGyQFJZpY7UlLkbBfR9DxgGbUXH6NzNE 2523
praxis/optimizer_prefix_vectorization.py sha256=qse2Vg0ymp5cxhGkVwtcIoOIu5xp6rhkpIgnxsygKqs 16997
praxis/optimizer_prefix_vectorization_test.py sha256=7yz2O6Luj_VXPTOAyWoxqx2lGeOtOe1RlPLIUAaRwoE 3196
praxis/optimizers.py sha256=Xj75Qh3Bc8moje7PtUNffF8QjPbl4GfySgbBKCt_qqA 113212
praxis/optimizers_test.py sha256=UAgx8E4nDQyMrXfDu9WCw_0YrWe1Y7hxiAgNLG65sY0 13883
praxis/pax_fiddle.py sha256=mik_KaJFaB5_2_FnGBkAeKqv5Up-qDdnjx4zmPjw9Ao 37119
praxis/pax_fiddle_test.py sha256=H5oP0D-DPKX-nb7gUWmDmajb_2UGmr8vEXGGhF5_8k8 41079
praxis/py_utils.py sha256=UN-3kNmQDjdHLJ6eDCTskHTC5vEA-yI6mBTuHcp66GQ 39434
praxis/py_utils_test.py sha256=OO9gymorfkHeHCZBzybqJOGzYXX9DgWJ9qqiqWaFEs8 18905
praxis/pytypes.py sha256=yV26TiM48hjPumrcVDFZKj-5UneopNrtjBuvYFSGAuY 5338
praxis/pytypes_test.py sha256=xMh5ca1M4zsv3qRw8wjGoHwTWhCeDrBhg5cxxU2oVos 1583
praxis/sample_decode.py sha256=XSFZ0i_uaMU4s2sEwlGS3zmE0JGJ8-m3vi4gob5xNGQ 76553
praxis/sample_decode_test.py sha256=eIpYF3zb6gfQNLRWf6JqCVHQ75y1qkHcZJ3yE23nGaM 26698
praxis/schedules.py sha256=YYieDTayzh0W6rTNqcXK5xUFENLoy73l6awdp_fSMiI 22153
praxis/schedules_test.py sha256=sm51wiJrXBbGJyA7IKGd6AnJE3dWmdFOKaci8C-H1Ro 21498
praxis/test_utils.py sha256=yykd2e0XYbzY9QIXn7DcxY0CgOyBVI5ttIwS6dpuZwg 18628
praxis/token_samplers.py sha256=ULDzoM3joaeeTRLkrMN9w0DjEJM6NHlWrYPQyyJVrB0 24670
praxis/token_samplers_test.py sha256=rBMQXU1h94A2K81UatkHmAYO0NkvZfWJymYVhcK-xqw 11225
praxis/trees.py sha256=hx2peAktti7cf5_U6Zj4gx5q8XuThdik4wHT_GKvyEo 8428
praxis/trees_test.py sha256=hjhQlrZdZBsZyJcsLxoHE_mG8WnZugMs8sA7tJPuoRg 12764
praxis/layers/__init__.py sha256=2hY8NGOVhakimqMd76pp09BjCAt7EfIB3B6YdN-1BhU 6959
praxis/layers/activations.py sha256=IM6PfE_21VXoIGbUFtKDbq3i4tGNWC_DQd8YqpBlyBo 5340
praxis/layers/activations_test.py sha256=OwXyYWTCLrAU4Lj0zfahVinBgsOy-yzzMIFL-Xek1zY 1858
praxis/layers/adapters.py sha256=8417K8nIXaGTQdRtv21skxN6FxiaciSzxXL8K33OPAk 8286
praxis/layers/adapters_test.py sha256=uQFpYN2w-RaPfaCdkBbr3xBXbytluysUlwe3QDGJXz0 6878
praxis/layers/attentions.py sha256=AixD1MVtqXEjK61F4_xjz22xNv_-jwbxifUgr0GZWy0 130505
praxis/layers/attentions_test.py sha256=1RwZsYQX5Hd_fD7rCwUoqDJc8wcT7eET-gmo8cJRNRk 88020
praxis/layers/augmentations.py sha256=xbYuLGu-WtVo6LKubwSZC_vxz5gFtPeR2nxbIq4YpJA 7169
praxis/layers/augmentations_test.py sha256=BrRSi_U4uVaCA7gvIDtL99agNIwJrB5-M2rWFPjvgH4 8903
praxis/layers/base_ops.py sha256=GDC0S5gC2ppdR2fR-15DZ_IZfIDOL8-eZXnfEEHWLZw 1673
praxis/layers/bregman.py sha256=MNSJXdN-vzvRBKR4P4JA9oOWe3rklR_TSev7AGfojDQ 12641
praxis/layers/bregman_test.py sha256=Vkx2O9iqo-2yHrHr1lSdOoE51_QGfzgEtvyfMsjmq-c 5181
praxis/layers/checkpoint_policy.py sha256=NspsvaV3XDZXre9y7znDAwElqV84FWuP6iK3sKoE8AE 4889
praxis/layers/chunk.py sha256=ZrQpwlrkzpbSvu5KGAsJ_fQ3D2pMMT5KwiXnoVWITFU 2220
praxis/layers/chunk_test.py sha256=ERTbcfaBfUn0-GVFIQGt8MZNnTVpr0oW2M0q9DM_B-w 1472
praxis/layers/conformers.py sha256=hfPgzQ01nhlQzxWKgZ6BKqQhNoCWVltNfViS23lJhwE 19084
praxis/layers/conformers_test.py sha256=Xmxq5IlEmmdZvC7AytXB70g0gzM33WNMz4ketRjmpK4 11395
praxis/layers/convolutions.py sha256=6N4RhKE-W-l1LB8JNP6ok7Kgd8H05QmX7RZhSOi9COE 34498
praxis/layers/convolutions_test.py sha256=WqAufuzz-3QSnQfqENwSu0gB9hR0O9oIKdNtZSh-OoY 19012
praxis/layers/ctc_objectives.py sha256=duOXtTSKvDJlEJt-Km3c9vW_fkgbmuB-XVR3IM87PWg 17834
praxis/layers/ctc_objectives_test.py sha256=11NWxLHpeR4qYQw_K2mpVNV4lv3fhX6vd4MjrXZJWHw 13365
praxis/layers/einsum.py sha256=F0RZhqR37NetfSsRUAWefboiYbAM1R2My8WnsysLNRc 3364
praxis/layers/einsum_test.py sha256=Nb9zDhaFvugd6u1bDhqNlrezVhjfdWYI_VkwfrAFEH8 2143
praxis/layers/embedding_softmax.py sha256=BFlLSDZ9a7twW43GNfBlLYpEcx9HozlvVahzGBDpD6A 50399
praxis/layers/embedding_softmax_test.py sha256=iv5_CQe_yZQMm3wX4F1s-Yty4xzQTZdnarb2zzaUwmA 39419
praxis/layers/flax_adapter.py sha256=_r1Y8uuJya9DIJiYipEhej0RgKkbRuRyPXQuUaynUY0 6733
praxis/layers/flax_adapter_test.py sha256=coT2ZPre-zwxuC79fIUmE0o9P2XOJYLyVRUMD0WDtlw 11273
praxis/layers/frnn.py sha256=my8USYXrawXJDlY7QMPzFnDCjympyKZzCj-PkyFYOck 17194
praxis/layers/frnn_test.py sha256=OliYKor0oz3LN-IChLhLQXnBpXcjFId70lFtz_u3HOU 17737
praxis/layers/glam.py sha256=32VwHN2BxigSL-pMr6ohYFc0eAVKKno5-Swz42R-3V4 13988
praxis/layers/gpu_fast_attention.py sha256=YhjhqEkuYg1T3NShG7kRcbHtJHBPzqoJ52UI4Rf3Fcg 17268
praxis/layers/grouped_query_attention.py sha256=Z6Md0I8BQlLEaTY-klc5mkhBhz5g7PwyUQX1EfMIgIE 16941
praxis/layers/grouped_query_attention_test.py sha256=WDPEO32b-hrKA5_I0NpQkn3Svw7MhZTdteLe_AD6-kc 6067
praxis/layers/linears.py sha256=hybBySc3jBuaPDNxFugkodosZreY5NG4G1RV_XT09G0 18696
praxis/layers/linears_test.py sha256=jU30PcXpzYVP48gtCo4yE5_5bvzqC4AJdKNCEysnwh0 29853
praxis/layers/losses.py sha256=ktTuvTywNtK7trNGOC13f3QO-Mvroctwux0vhxj39xM 3826
praxis/layers/models.py sha256=FeG7OxT0r-SbBJpyKy2gCkb-ozGFZEHms29v0HYIWxM 78696
praxis/layers/models_test.py sha256=xmBju_29cLrx2yOnc5k1kaTL3ZFyT6siO_sIbc8UZZA 57477
praxis/layers/multi_query_attention.py sha256=zp9HJ1RxOmDpi9krbwcgJDgNq9MU-NHvMGic5jfCC0Y 65639
praxis/layers/multi_query_attention_test.py sha256=2gScrXW-sEuUbeAQN1WYmcF0buGyYHOIQvNhiFORfSg 28992
praxis/layers/ngrammer.py sha256=aXre09FgYaVLzXmSY8bbXjqjFaTr4zGVtjT0BKkGv-k 51858
praxis/layers/ngrammer_test.py sha256=A5Zv_vSjMbIlqyWuFG5lwdnty2kEV1fI9kZLHfotVAo 26513
praxis/layers/normalizations.py sha256=N5JhouAxCJpV5g3sBSSrKwFP6VZa1gSs5aMGtwgIMpM 22231
praxis/layers/normalizations_test.py sha256=yHnYhjzKiibfgeX8BeKtJfFWM6_XwtK4qf3cMMbXrFc 19447
praxis/layers/pipeline.py sha256=gmVfmKBZhHORQa4O8tDQ7yu1hmAk2i8Ufc_ce1zEyFA 48753
praxis/layers/poolings.py sha256=Rutv1xrIYgGVs-XCGnrNpCiDy_TDZ9IKo9vNUj9RIVY 13518
praxis/layers/poolings_test.py sha256=Yl8LR_5u1ZN5c5bMifqzdgXZwT7HDXsjNTfh0inTxPI 9528
praxis/layers/quantizer.py sha256=AHMeRy-p_Qb4PzuclZvdQQ-b50IQuWJbz2cLIt2pSmE 16337
praxis/layers/quantizer_objectives.py sha256=ZhoGMBxYiaAQXyvO68iC8uJpDQU93NU53VTBIyTBGUM 3892
praxis/layers/quantizer_objectives_test.py sha256=YH8lI1fosaYE1PUl9R3TaOqZCWpUQPX2dhbrUgrvMfA 1690
praxis/layers/quantizer_test.py sha256=EoAwcJo0-vi2zQ24sPbXl5V4o2imSa3jQU6DckTyTcE 4351
praxis/layers/repeats.py sha256=SHl4zIUAjfURls7jHo5Rt8Jq6uz36afQLQ1S3tRW6ww 23959
praxis/layers/repeats_test.py sha256=YZZbyKEegkmceymXcO3fgpZXBf5eNfX0cq7iPjroWtU 16378
praxis/layers/resnets.py sha256=AO4itTc7kbCr15OFaXDUYtW-XWtHaIXKhLoY5jcNbdE 16250
praxis/layers/rnn_cell.py sha256=ZsQbZJy8XH7QdhoXF3daEyWySDufWxIGzCoN6HEoELo 17426
praxis/layers/rnn_cell_test.py sha256=cidbSZleSwlLi34TRWtCukALY99UKuY14pMqM44YCQ0 13660
praxis/layers/searchable.py sha256=BmZKDth0EXEbVhhcnz3A7g2HwHtHbPI8YiWzRyhm-ng 2290
praxis/layers/searchable_test.py sha256=F_XpvaGyYWaKfKq1fTb3UATowsN4er1g-KwsgeMfqTo 1910
praxis/layers/sequential.py sha256=8iPzdH44WmGabKEtLbXj6qEzZZeOwV7pGg1xQ5M2Wuk 1270
praxis/layers/sequential_test.py sha256=EhKQkCk3Egd04wwHSffMXtK1V5q2eEmc9lH3-xMyFis 3167
praxis/layers/sharding.py sha256=LBW7yJrKjhIYD26Sb_SPgMDfPizuBbvXnUHyA3mwOXw 4657
praxis/layers/shared_layers_test.py sha256=Qb-faxRUx_9JtHlulbLxfvaCLWtId7xSUYTSvgJtPqw 9571
praxis/layers/spectrum_augmenter.py sha256=MGb9GEkn7KVU9KHZ8Lk2R88A3aWKxrr7-89TuCTbF6o 10049
praxis/layers/spectrum_augmenter_test.py sha256=80dg-qJjpPmu8fUkP7YoOsPLWyl51NZYSjqHXvz5BFw 6714
praxis/layers/ssm.py sha256=-k72sWCr3zA-Gxgjr1zKNttmZbW3ZHQRqfUviJx4j84 13853
praxis/layers/ssm_test.py sha256=73G0-MLPvdZu4aF3o_uPgCP0KWKvhlx-fW75o4bZpSc 4454
praxis/layers/ssm_transformers.py sha256=1gsPBnnrbFoJtFWOCnJ2lM3Q0z9AAyeijPP0mpZpJfI 26821
praxis/layers/ssm_transformers_test.py sha256=7n6qdNGCbXI-El8-qrtKXN3C_92eln6e2G_l4G2Ftwk 5695
praxis/layers/stats.py sha256=uaCXihIB-XcUq4YouZCBJuLU2XMd1Fn3_IccpdL_nJQ 1537
praxis/layers/stats_test.py sha256=aAhFRqh-YGqEUWseyRfRDpPBBzaOpW_-a263kI5JI3I 1258
praxis/layers/stochastics.py sha256=SkfgpAYntDMgaYtQItvYL7cCu8B0DlDmytVSLw_m_RI 5159
praxis/layers/stochastics_test.py sha256=5T4keo-a2J4KG6leOn5XynYgma1GerHjLefpIpfRT_g 4517
praxis/layers/test_layers.py sha256=meIGQI033WbhbKcYlKilfIAIjimfviUUic1WBLUsl6o 7898
praxis/layers/transformer_models.py sha256=ZxwVFsinU-oejxuhXLlcWn8vY9RJ8EWjX6Lvplx7NHc 80038
praxis/layers/transformer_models_encoder_decoder_test.py sha256=MuoOoJgLsEMetdNAUE_cvk5mZ5qmt3cDtTpHSPgXaUU 9529
praxis/layers/transformer_models_test.py sha256=IVSgjjWkH1vsnbV1UgmzI0IDcmaj2luP0vF8upyRz50 58376
praxis/layers/transformers.py sha256=KHhSYcz3J-Eu7DJHY4OCg66R3A559OoEo2rP2_pJg2c 93733
praxis/layers/transformers_test.py sha256=bl5Vik0UAt5NZ-73lr3pfZHKduFkUUVJAcMhsQWo3MI 67654
praxis/layers/vanillanets.py sha256=UvjPtbiJdaj2VK81LBLEHclakhQPTBTk2osQYfkPsvQ 10068
praxis/layers/vanillanets_test.py sha256=2mT68jfnoTsOeRkLrT-2E3O9oNklucxnSlshl6AhVLc 2836
praxis/layers/vits.py sha256=vMtTKWnKLhQZjiXNIFw0W_lnyiNZCTjFGG2YW8qk_XI 18433
praxis/layers/vits_test.py sha256=FPB3hLOv0Q4d5-3hIhA389nZpMqZyeTfuFf1UFeAcmU 9661
praxis/layers/chain/__init__.py sha256=K9-2VKZPc1r62iDuzeSgENX8Xysnjq9R75CAizF4prU 1343
praxis/layers/chain/chain.py sha256=BPpiWrr67z72KwKKe-wbYHKSG7V_gOhp1hQyaDQ-RA4 5210
praxis/layers/chain/chain_extensions.py sha256=0tFGP9ME3cNwxveYU_a1UoTI9LKI9nEnJsHdHG41TXc 8837
praxis/layers/chain/chain_test.py sha256=xaysnpVH7D7qSHnRWbPPiKq3y2_XJuWs_d0-krAlegU 6913
praxis/layers/injection/__init__.py sha256=5epthk7t7Wb4jqn0zlhJZDt9l0jFD0V1p3_PuBRC2lA 710
praxis/layers/injection/fp8_nvidia_gpu.py sha256=40xT-6-dW9NcFKp07qiUfXlpNn_S3qcDfhGFY9LeXPU 3718
praxis/layers/injection/fp8_nvidia_gpu_test.py sha256=XVwM8bTLTn9O1fDsrpTrTgom2Zl8i780D7yffg9ivg4 3603
praxis/layers/quantization/__init__.py sha256=42CmAtm6P7l6daATWj4CGHW_SNqoVy34FMRIXRQCl4k 2173
praxis/layers/quantization/attentions.py sha256=DEecoqFdehEseuFxSGFJXucs7ZBeg3NFnEiESsVasqU 38463
praxis/layers/quantization/attentions_test.py sha256=t5jrr1FAVSfPL4NyoH0WR947ap-eRai20ezCGWezNio 23701
praxis/layers/quantization/automl_select.py sha256=cTdo4CCJyLSdSETPPC9Zy4HFYY1A6rfki_sBc-wNNZU 3101
praxis/layers/quantization/automl_select_test.py sha256=WgEWWQrY54CI0naaNm_h-VWKyPHt8phtfE61YPjW6Uc 1937
praxis/layers/quantization/conformers.py sha256=3pf73pv1OO6TCdYP-tH8u0t-rr3ZfPZDfzzaMMWJIHc 2703
praxis/layers/quantization/conformers_test.py sha256=KePgABNvbyjmTufJ80K1imTAgS9ig1uZlJodfVMN1ZM 3217
praxis/layers/quantization/convolutions.py sha256=azQ4J6CiRya4JcejIQerGo54hmXrR-QPZVhvjdk1Axc 4105
praxis/layers/quantization/convolutions_test.py sha256=ss8Pz2tGlsX3o0tWI0kDQIDHGcFegBzcCkWlUVbs7jc 3698
praxis/layers/quantization/einsum.py sha256=sa83iEu8G69vCmiMScu-iyo4_ESl0QyTOxe6Oao44nI 3349
praxis/layers/quantization/einsum_test.py sha256=nttDQe-UiPVliJcnYmBPm9MXEeMzgQZzziknmR3z170 2338
praxis/layers/quantization/embedding_softmax.py sha256=1p-cWjAcpJrTIZQMwQ1lKRTDDxWWbxHwAkPKmMzKNYM 23612
praxis/layers/quantization/embedding_softmax_test.py sha256=2_krJMmayqluwi9k-EkP5dbTlsHFjwNuZjR-hUisWyY 20031
praxis/layers/quantization/linears.py sha256=9ekV9C8k8rvT14ub8bvk97pkZf5VE5_Q6J65szCDUDg 18061
praxis/layers/quantization/linears_test.py sha256=mEAfQ3poKpb_KERtZtIQSG3HKQzaHMVmvk2c6opL660 27071
praxis/layers/quantization/multi_query_attention.py sha256=G4jsmBSqtIwtNFQdVxj8FZmlISrmJA81KetdWzTv-s8 7833
praxis/layers/quantization/multi_query_attention_test.py sha256=9Nw7KkDyAxGkuui1fP5D7B65wiO64A_v1ZuitscZArc 4889
praxis/layers/quantization/ngrammer.py sha256=vQdX5iuZpkkbYUB9pIUqUZGub18Zqx0DkhY1srm9ZsY 4853
praxis/layers/quantization/ngrammer_test.py sha256=iPecN4udH2yXjdx1ubGuj9_8Rqcr1XURGhbLSETCY84 17527
praxis/layers/quantization/operations.py sha256=d4-UUSbOhUIRd4mSFoxp2cRkjqmIVdVQj6PjABI6c5w 38313
praxis/layers/quantization/operations_test.py sha256=aJIA_BnJzNf1CXB6Wl29Oblg9b2Izjm12MObNjKIJa0 37067
praxis/layers/quantization/optimization.py sha256=H8MygGFt9r_dK7ydIhv153eUvGlVmoXCzjmRkhnSZvQ 5601
praxis/layers/quantization/optimization_test.py sha256=U9_bSMmTnJXzj9gQTrcgQC60jL-WloQtBbeLk7kPXI8 2479
praxis/layers/quantization/overflow_check.py sha256=5Hb3V6bSYczYZczl8MoTXM-WPc0pAHNFm2YOWO224Ug 5388
praxis/layers/quantization/overflow_check_test.py sha256=2-JnLZRPsK4j6v7mPDmwWiGGHq-wT38VwieX5rbhmQ0 6063
praxis/layers/quantization/quantization_hparams.py sha256=RfkMuN-JY7NtZmGS0O1wZkeCPDbZIycVz3Eo7LzNz1w 9082
praxis/layers/quantization/quantization_test.py sha256=jYwZn9VXfO-CR1vHB7OyfmvZcFk7CPHJ3hN0hWpMVfc 5626
praxis/layers/quantization/quantize.py sha256=2YkBK8gg4sF9GvOvDVTzkjIOI9jDuDA69rDYls_IvWA 40176
praxis/layers/quantization/quantize_test.py sha256=JqjSxRXWpOPqq3fUz7PiXl8hGWHL20JYyzfMgBLQjVs 18578
praxis/layers/quantization/quantizer.py sha256=r7bggk2iKIIyV1vBXAY1pBWf7TVMygLl1Rt4ilH-D1M 29590
praxis/layers/quantization/quantizer_test.py sha256=iOCM2vmfid1Xp1sTsfbSjfQpjAz9vGc3-D3Etn9hxgs 23806
praxis/layers/quantization/searchable.py sha256=itGSYWW6si2Dw6_ZyV-2KqrfaEdYDcwhPdIAWDkNCss 3965
praxis/layers/quantization/searchable_test.py sha256=vhfnGX6nh2d7UYoo0XAIOw15CzCFWzeoCGkcer5C744 3790
praxis/layers/quantization/utils.py sha256=yyZWSw4nqLmEmzTlC_dug4Wn2SQY0m4GeJeSuLDEJR4 18029
praxis/layers/quantization/utils_test.py sha256=2MuDULUevdFYt-UyYUBkqdPfP2PVcmw8uabm-d-3XBk 10211
praxis/layers/quantization/sparsity/__init__.py sha256=HB1NQ5f1jkelS-Vycrf4qYCo6foxYs9gp4umxo8Rlqw 644
praxis/layers/quantization/sparsity/attentions_test.py sha256=rWItjEyNE3XzXJgUiy7HlpeJo4JT3RQCnlRgqADqMZk 15376
praxis/layers/quantization/sparsity/linears_test.py sha256=btNGUNLbbCmWMy8sDhq9hxDz00LY-sYCAb325MGq8Cc 10473
praxis/layers/quantization/sparsity/sparsifier.py sha256=phsyfdFhkNL2jA15CNowkV88Kf3TJcIAfh3OZoffJCo 19155
praxis/layers/quantization/sparsity/sparsifier_test.py sha256=qrIXJDrWo6JY74GwiyBPH6Hj3FL5DL7VjgkUm-j8T-g 44642
praxis/layers/quantization/sparsity/sparsity.py sha256=-dmnH0QWu3dhJSqKgVda2ocbDXzTqBOP42qILj0h3Do 13598
praxis/layers/quantization/sparsity/sparsity_hparams.py sha256=XNezO_f1WFKLPlpyHrQdAxDLQYDZW7zlqjJnlUSh8dE 9407
praxis/layers/quantization/sparsity/sparsity_modes.py sha256=S7tUq2U6cFB65UEXb8PK_q7GgYbw-LkfWGXLB58i7zQ 5898
praxis/layers/quantization/sparsity/sparsity_test.py sha256=GdhEku5d11RV45ToIA5r-js52Vw_cdUr6QavPAbLwNg 19317
praxis-1.4.0.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
praxis-1.4.0.dist-info/METADATA sha256=X2akryVtvH-cmqHviMYX00Je-72Ryy27xL7QxjFNaLE 1218
praxis-1.4.0.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
praxis-1.4.0.dist-info/top_level.txt sha256=jQcnqGC96gVr2kN5zV78HwY1gNcsZ-8JnoLEWogxu1g 7
praxis-1.4.0.dist-info/RECORD

top_level.txt

praxis