e3nn-jax
View on PyPI — Reverse Dependencies (4)
0.20.6 | e3nn_jax-0.20.6-py3-none-any.whl |
Wheel Details
Project: | e3nn-jax |
Version: | 0.20.6 |
Filename: | e3nn_jax-0.20.6-py3-none-any.whl |
Download: | [link] |
Size: | 166582 |
MD5: | f06ad5461f4bdca968935d114aea37a6 |
SHA256: | 0f4dcd124695274608270a8a99599141c542c2317f70921ee0bdf35818a87c20 |
Uploaded: | 2024-01-26 08:16:30 +0000 |
dist-info
METADATA · WHEEL · RECORD · top_level.txt
METADATA
WHEEL
Wheel-Version: | 1.0 |
Generator: | bdist_wheel (0.42.0) |
Root-Is-Purelib: | true |
Tag: | py3-none-any |
RECORD
Path | Digest | Size |
---|---|---|
e3nn_jax/__init__.py | sha256=4F5WH3-a93_5-RseauD8BnaqRUPGfx6Je0ZUnRvsQq8 | 6067 |
e3nn_jax/equinox.py | sha256=OyPkxixpjJcKXgyJbf2TYlKV4-pkUu-1Y3Qve9mg86I | 154 |
e3nn_jax/flax.py | sha256=qgEglVoik62gq5aFfezTB6jbcQQT0UKgu8ZzmljPym4 | 300 |
e3nn_jax/haiku.py | sha256=K8-0iOm-tN6SHqDJW0XLlad8hfBZpwkC6uwhcRwJPQc | 634 |
e3nn_jax/legacy.py | sha256=IHM5v75NI3NfM3kCLqGlgZ_8IS79wQjcD3o6vtyT-jg | 335 |
e3nn_jax/perm.py | sha256=kIzTl3OeQLgJywufnyzfDTN73VllmSfHHFdm3up39R4 | 2339 |
e3nn_jax/utils.py | sha256=HZLVdbqBfaIbQb986Dzna1oEWghr7vRW5m5jPhF3EzY | 300 |
e3nn_jax/_src/J.py | sha256=AOI1YIZT02xt-LUiaeTqlnO7H8ehim9GaH6ZoaH_jMY | 40652 |
e3nn_jax/_src/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
e3nn_jax/_src/activation.py | sha256=PGN_i1KBv2qDmz5labchLH4qDpN4UsxTQze0rYm_UHg | 8069 |
e3nn_jax/_src/basic.py | sha256=86pVV2s1vfPxdJb-9fl7kc3Up2Bjq9XNrF2mxFngLvg | 21916 |
e3nn_jax/_src/config.py | sha256=iHDwIo5_ZtLlj0XCwZfSd_WPv2Rd9-jfhM7knA5J4Us | 714 |
e3nn_jax/_src/dropout_haiku.py | sha256=ag51vgJuDf9KjQyayukYVdMxgnkyJSpw52e0im0KSCM | 2504 |
e3nn_jax/_src/fc_tp_haiku.py | sha256=rlZNk9QqshrazS_7XugSIp48Dp7hFQHN4Mo86GmJLFM | 1908 |
e3nn_jax/_src/gate.py | sha256=R7Ff3litDH-HreELE_O6CqqM8PMxYo_CYcY-GmWCm1w | 3266 |
e3nn_jax/_src/grad.py | sha256=ANRwXNjXYpFve51UCG6xcr77kUClaNKHQ1z0Le94F4Q | 4380 |
e3nn_jax/_src/irreps.py | sha256=5-4NY0uUDsw_AjAD9yZmGS3JjJC3eBJi9-cm4CH2Uq4 | 32813 |
e3nn_jax/_src/irreps_array.py | sha256=qA5gaAMIt82Gf-5AqjuHcLxf0S703qsC7Y4jLfvKnKI | 51291 |
e3nn_jax/_src/linear.py | sha256=hPUj95CzVRJguFqFjSnVdKyIsnLZCrp1Tu7mNHga9Zg | 12351 |
e3nn_jax/_src/linear_equinox.py | sha256=TKZHhNRPtEomAshJflZPQ2NsBvqEe0mCnGr014_6RL8 | 13141 |
e3nn_jax/_src/linear_flax.py | sha256=ssHPh9KGRl4YydjHOL3vbYw7FLUHBtv6qJ699UY-qa4 | 7499 |
e3nn_jax/_src/linear_haiku.py | sha256=jSbOA2hx1RvJFK33xH7jYym9mwO0b47oWHqT45dNFns | 9347 |
e3nn_jax/_src/mlp_flax.py | sha256=ZREjZUnnWsgo3-4Kjnr98J2yp2wA1Zj3ozEjh5iQZ0U | 3593 |
e3nn_jax/_src/mlp_haiku.py | sha256=Dxc5ATOHzX16JqACh4NMl60pWDs5sXPt5mSZb-Phe-8 | 3816 |
e3nn_jax/_src/radial.py | sha256=iDYyP2tWA8ulbZxUx5atjJ-h5YuQZ9M5q6s1JEshokQ | 9793 |
e3nn_jax/_src/radius_graph.py | sha256=OGZNhE8PRij2DPT-220qqIH8stHKuF83hIIZVy9K3RQ | 1697 |
e3nn_jax/_src/reduced_tensor_product.py | sha256=K8rPFs868yleRxmWHtWf6Yfpry-fvZhIsZHReXiV82I | 27596 |
e3nn_jax/_src/rotation.py | sha256=M_Tt0P3-Sk_JQQP-UnlwN02HsQTHN4TGj4IKk2p3KFk | 23669 |
e3nn_jax/_src/s2grid.py | sha256=Caka9P9OWEy0DbcF3RkFc-kOH-zrXEH-5k2OTniqqCE | 55530 |
e3nn_jax/_src/scatter.py | sha256=PiX6RqF2xGU2zuxQtEVl7NL4okHKuVU7MKX6mEn9Oto | 8638 |
e3nn_jax/_src/so3.py | sha256=ZZaBWnRl2gfoapwjDKbQciK8D44prXtmlmCRAM27-TA | 1789 |
e3nn_jax/_src/su2.py | sha256=nA2rymRlZD0UTRXPzjdhEQc96Fnp5hedLFKXsa78ros | 2412 |
e3nn_jax/_src/symmetric_tensor_product_haiku.py | sha256=O5cs3Whsx7tbzr6rcguhCYZVg6JUkP11K-3dSZVo7RE | 5633 |
e3nn_jax/_src/tensor_product_with_spherical_harmonics.py | sha256=dMTtH4XVGdZ_wQt9bNdY-H2GY-IVNw9Mv3Z6smZacu0 | 5601 |
e3nn_jax/_src/tensor_products.py | sha256=B1dwwbVoOT1R8BldOgXLEykv9SPl9QCoCF1Os0_LI7I | 14996 |
e3nn_jax/_src/batchnorm/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
e3nn_jax/_src/batchnorm/bn.py | sha256=7a6jJXTX3_JlsIK5OSeMUVCxaASu8nJk4X5BAEsQ8YY | 4829 |
e3nn_jax/_src/batchnorm/bn_flax.py | sha256=TyOFCyDiqEPJAskw8fgBsbRlp-fc9de7ihvJ6Kenb3E | 4651 |
e3nn_jax/_src/batchnorm/bn_haiku.py | sha256=rmq2K1ym4IJ2gbTXd5iX8LuIdCzc_bULAY-Q2tjtslg | 4086 |
e3nn_jax/_src/legacy/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
e3nn_jax/_src/legacy/core_tensor_product.py | sha256=KXD_FdqQadtV35f3RcJvmXy1J9xjOUdrlucPiBqAoqw | 29283 |
e3nn_jax/_src/legacy/functional_fully_connected_tensor_product.py | sha256=DPEQI3J7kUo09Aylmti-xZ7fiiKNRE8bRwnwXBe278M | 1135 |
e3nn_jax/_src/spherical_harmonics/__init__.py | sha256=sILyb_JOsNUMzd9zFccdxip2uFUuz8dpcVwYoikzbzQ | 9205 |
e3nn_jax/_src/spherical_harmonics/legendre.py | sha256=8yNqc3HF_B5VjNthpaIJjRntubrZSGursRcTlvoVHkI | 3240 |
e3nn_jax/_src/spherical_harmonics/recursive.py | sha256=hPqVPaJEWNsjNKuQS9Cgcl1G1SsVa275SdQ1NJadfA4 | 3497 |
e3nn_jax/_src/utils/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
e3nn_jax/_src/utils/decorators.py | sha256=IgtJoikOJ9gRCFHV0rWtTwCt9PFaEQYLgTIWkXKbXGU | 2027 |
e3nn_jax/_src/utils/dtype.py | sha256=yeC6RXmHTZca55-zHx0fR8QraOkGbTCfzMm6da0HL2A | 435 |
e3nn_jax/_src/utils/einsum.py | sha256=od94bR9Cog33KWZr-mA20ETMBSvo_fihbbFJ-eGhG3w | 443 |
e3nn_jax/_src/utils/jit.py | sha256=vrK0xiLRIU4B8FapEGPFrDAldSz0qloik_aHL67w2rU | 785 |
e3nn_jax/_src/utils/math_numpy.py | sha256=wQq_09sDY6ycVWxbN3RQ0-OAyc9O0Z9I477TK1FJumo | 4908 |
e3nn_jax/_src/utils/optimize_jaxpr.py | sha256=u3b_NshNm_tw688FTAfBPCE8tiS9p2_4Xa7BoZCxumA | 6864 |
e3nn_jax/_src/utils/sum_tensors.py | sha256=jZrzfCvGpF2UXB2nICd_xrnO3j1sck9Zf-wt3eXOYX4 | 360 |
e3nn_jax/_src/utils/sympy.py | sha256=wC8aVQVUzwjY_x0DN_jQ2mZBHfCe6_E9uhODiTbelIM | 486 |
e3nn_jax/_src/utils/test.py | sha256=j1E5Eta0OpjT1SrIFOImnerf-uVIWokFcKOKsY9seXE | 4368 |
e3nn_jax/_src/utils/vmap.py | sha256=SrFJq4LACz84jyuV8W2J19EesgbourO3X8BEoBL1SWk | 2347 |
e3nn_jax/experimental/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
e3nn_jax/experimental/linear_shtp.py | sha256=lbZsk5NQf271O8xBruflX-ICVSuLZ0wVSpok1s6VYaQ | 9261 |
e3nn_jax/experimental/point_convolution.py | sha256=bTiK_kl6XJQbTDeQavw4cFR8CiK22SfTHQM8yqWYnVU | 5864 |
e3nn_jax/experimental/transformer.py | sha256=-tW-fvPFnAn1cd5zQ3BIPTeqUPPdTch1EZkUXVhho-M | 3325 |
e3nn_jax/experimental/voxel_convolution.py | sha256=B86B_6Pwl0rMF4ezgb1rSaGrK0ZTbZs-oEAR5hp2RR4 | 8526 |
e3nn_jax/experimental/voxel_pooling.py | sha256=uOsEvybDbH0ncLJoaAqpAg2OKz7q6AH88wkDCnD3OOk | 4123 |
e3nn_jax/irrep/__init__.py | sha256=OWy5BXb2M0sen4ZghIuDtbAUQGxPXBXkwMgtUlkXpGk | 901 |
tests/conftest.py | sha256=Hz7NenjVMzP93h2alcHYSocLK6iE57x7okK-MnSP3uU | 678 |
tests/noxfile.py | sha256=kMYAT7I2d5emAjjVclAmvBfvT-6_-n0xwxIq3j_ZhZU | 259 |
tests/_src/activation_test.py | sha256=OTdop3VWckfn9UDr6ChkDbup4UcY8nwdfwowtmeWYJw | 1071 |
tests/_src/basic_test.py | sha256=NfzmKXVf7sbhGarJyDr7m7frePLxd2w95WnoJe92SBc | 3187 |
tests/_src/config_test.py | sha256=qOlBeWrqaqHZ_M4RgaE1CUGNhOORMNSgCVi7ayzKWJA | 914 |
tests/_src/dropout_haiku_test.py | sha256=rv-tPr3TLsxWmvZUyGgdjknfbQqi9mBDuLuIHsAeboA | 718 |
tests/_src/gate_test.py | sha256=XVARwPwS8Hqim9FV4SY_K4tOP99QoyJGPpbbtWjRxpM | 709 |
tests/_src/grad_test.py | sha256=sZwrdxDr69k0CzOSr5gIM8TNWeFiwGlRzLnMA-XWCAY | 1793 |
tests/_src/irreps_array_test.py | sha256=S61q_zHyzuFFXq4yGAVhk-HtiHNTmbKjK9smqR4Bkic | 7834 |
tests/_src/irreps_test.py | sha256=KvOXQbX8hVHf-Y8dv9T3Y-7gC-P85fjqbRt9pL6d80M | 5194 |
tests/_src/linear_equinox_test.py | sha256=e_6Q8dKKKbb7Y6kRdxvA5W9iGnMoGFtCA7juJgRpI2c | 4590 |
tests/_src/linear_flax_test.py | sha256=_sset6YVdNchfK555bQd2uF-eP-u72bl--BmZXKiZ14 | 3723 |
tests/_src/linear_haiku_test.py | sha256=sNyoKAv_zshrJeCwI3O2ARGnf3tEYh0uWhUH1wp_COI | 3509 |
tests/_src/perm_test.py | sha256=RjpwXklnu6ije9GtDz4plkhiGcDkLaHQKL2_hAJX8DQ | 1572 |
tests/_src/radial_test.py | sha256=ZKWPMV6UhxW8MpC9JU8Id4f-v4Q18ClhyuG6hPa9qLw | 2551 |
tests/_src/radius_graph_test.py | sha256=Y5-MFCDOUlDm-wAyGDOBCHIp7urulmK_AWMJGmWeVp4 | 653 |
tests/_src/reduced_tensor_product_test.py | sha256=j-tkNS5YtnCg1BkTdvoKotHdltiaMTwkHsAHvZ-O0Dg | 5614 |
tests/_src/rotation_test.py | sha256=aJ71uoml7Jd8Uo2ntOCglQVM42VFAdbPSwAPc4KxQeg | 6713 |
tests/_src/s2grid_test.py | sha256=m9EDfKZ2sSshdGigDEbTiKQdQu261PMn8xbH8M24w9o | 18883 |
tests/_src/scatter_test.py | sha256=YLELPDQ1-3nWtB_-METICHqwQSe9dn19Upt3DPwFjSg | 3659 |
tests/_src/so3_test.py | sha256=xfbtic5I0edBQM_LW7rdVKq36VEvCcsFdMhXP6Tjgsc | 2844 |
tests/_src/symmetric_tensor_product_haiku_test.py | sha256=HtkrsUnR9AXwR4S0zF_YX5AOaLGdASB6GrWHprZCNh4 | 487 |
tests/_src/tensor_products_test.py | sha256=Y7FR7OHz_aDDSQB6XIrnGt6n5bidRjkCL_g6BPKSU5s | 5640 |
tests/_src/batchnorm/bn_flax_test.py | sha256=evzSkQWWJ0fPG6WLtVBCd5msV5c7qMIf2s653wRWWMQ | 4226 |
tests/_src/batchnorm/bn_haiku_test.py | sha256=iti2p51qzljHfQzLVMe3P0Ax-oejddIXQ43cM6c9ihw | 3674 |
tests/_src/legacy/core_tensor_product_test.py | sha256=GGhsEqatyIZsC4L2pVAKglneQLwS22xYoB34NpgHBkY | 5609 |
tests/_src/spherical_harmonics/sh_test.py | sha256=FrnSMpoGulywBo1U11mcef0ayfv_quvNXOAK4xSblHs | 5409 |
tests/_src/utils/einsum_test.py | sha256=OGvs70byujtAq-T3NQQv3nlEGXwscmklv1638kKQCO8 | 863 |
tests/_src/utils/jit_test.py | sha256=Bpoai9tKX1s26hOyc5B3CFOn8-ZgjDM82HTcpbAlko0 | 119 |
tests/experimental/linear_shtp_test.py | sha256=5IT2sAatVz5vLBRUGrl-dmRRLN9rRHy5zyDSxp0ixTI | 1634 |
tests/experimental/point_convolution_test.py | sha256=f49Sbgvxxg7GyRvSELU5FiC3kvYKkSO44lf1lMGXJec | 1675 |
tests/experimental/transformer_test.py | sha256=8ZeUuJzEffo5ESAPr3837fA8Z0x_NUaZUHZNBMTMIdU | 1668 |
tests/experimental/voxel_convolution_test.py | sha256=QW-T7NRB1K3WSF4mi7lYJ2ixxzqEI1py94FQrtlzLrM | 2563 |
tests/experimental/voxel_pooling_test.py | sha256=ECNh2I5-PgjwmH0QWUKV8Mp-lEpD77lRygqZCkBe0ho | 857 |
e3nn_jax-0.20.6.dist-info/LICENSE | sha256=yEtlBjogbzeC3Waw_DSwY2HE4RwHLdiF8K31yNF_fFw | 11346 |
e3nn_jax-0.20.6.dist-info/METADATA | sha256=21x6vtJ5zrvpGcDk6q12482CzR2ns3-1yM0gJfsDwDA | 16559 |
e3nn_jax-0.20.6.dist-info/WHEEL | sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM | 92 |
e3nn_jax-0.20.6.dist-info/top_level.txt | sha256=0m31RxjShR1n_tWu9pILzF_CaGGPddkhyL8loM7WoMk | 15 |
e3nn_jax-0.20.6.dist-info/RECORD | — | — |
top_level.txt
e3nn_jax
tests