e3nn-jax
View on PyPI — Reverse Dependencies (5)
0.20.7 | e3nn_jax-0.20.7-py3-none-any.whl |
Wheel Details
Project: | e3nn-jax |
Version: | 0.20.7 |
Filename: | e3nn_jax-0.20.7-py3-none-any.whl |
Download: | [link] |
Size: | 168938 |
MD5: | eb3e461774db2361f0af7b994c0bec32 |
SHA256: | 8a1b6f0c00feed565358fceec4c916a623412b6328945fce272f0c33e52b4bfc |
Uploaded: | 2024-08-14 05:15:15 +0000 |
dist-info
METADATA · WHEEL · RECORD · top_level.txt
METADATA
WHEEL
Wheel-Version: | 1.0 |
Generator: | setuptools (72.2.0) |
Root-Is-Purelib: | true |
Tag: | py3-none-any |
RECORD
Path | Digest | Size |
---|---|---|
e3nn_jax/__init__.py | sha256=Mjanio2rSa9AqqjI5RdMxV46Fqt39WaDeNWoN01J0Bc | 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=IBVbIIr79bpdAG1mEqFeLzr7kixKWANVAMB-TeY6tE8 | 22235 |
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=u3d-3jEdPXgXkeybHtuLeGGnT1N6f4p79cpe_h4DF-8 | 32815 |
e3nn_jax/_src/irreps_array.py | sha256=4WN5nT-URaBJu8s37wEOrBO9_FayKzrQ_nCV8VQNH94 | 51550 |
e3nn_jax/_src/linear.py | sha256=wfgT1-Z0Tu1mIMOxJdRbZLJlIIUEBB1nOKbKI59swvY | 14527 |
e3nn_jax/_src/linear_equinox.py | sha256=dIrL-xutci4YZrPT7RelY5bB7y87W_zSrx2zApsKymw | 13142 |
e3nn_jax/_src/linear_flax.py | sha256=QlbhG6-iVjMp7I60NfuTpOmleFlTITXoyzKByjRoNws | 8018 |
e3nn_jax/_src/linear_haiku.py | sha256=GBbsA7XO-PV193PnrFhv7siPIA6fHaD88683iS6lUqk | 9636 |
e3nn_jax/_src/mlp_flax.py | sha256=8DsuO7kVDBSAJMhXCzmUJsoLc3KzzPd-aMPyYfGYr6k | 3594 |
e3nn_jax/_src/mlp_haiku.py | sha256=Dxc5ATOHzX16JqACh4NMl60pWDs5sXPt5mSZb-Phe-8 | 3816 |
e3nn_jax/_src/radial.py | sha256=ZhILNBOexuy-M7JyNPys5YrarADfuYspAnfkKlLB8sk | 9825 |
e3nn_jax/_src/radius_graph.py | sha256=OGZNhE8PRij2DPT-220qqIH8stHKuF83hIIZVy9K3RQ | 1697 |
e3nn_jax/_src/reduced_tensor_product.py | sha256=K19bBUp03ExVm50-WV0ljIZvWGbZ259TBj693i6ppJ8 | 27597 |
e3nn_jax/_src/rotation.py | sha256=M_Tt0P3-Sk_JQQP-UnlwN02HsQTHN4TGj4IKk2p3KFk | 23669 |
e3nn_jax/_src/s2grid.py | sha256=GuN-nL1pEybbMq7on7Yh5ZJ6G6l7QzSgtRr8DCX-ckg | 55531 |
e3nn_jax/_src/scatter.py | sha256=PiX6RqF2xGU2zuxQtEVl7NL4okHKuVU7MKX6mEn9Oto | 8638 |
e3nn_jax/_src/so3.py | sha256=Gf_O3kvnAr4ni6dCD_HFv_b2-VN6WTz8YUxL35XtDDM | 1933 |
e3nn_jax/_src/su2.py | sha256=nA2rymRlZD0UTRXPzjdhEQc96Fnp5hedLFKXsa78ros | 2412 |
e3nn_jax/_src/symmetric_tensor_product_haiku.py | sha256=8yi2sr_4eAJRMzxqMA0_TTdtGQE5QUi2fZ8iseeR4ys | 5634 |
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=MKvnnaj6mtXVtRheNKe3rZ8kq8rVoaMHUvz4eCCznD0 | 4657 |
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=9MIK5jjz194ENbKCafZGh89Tt-BhnqH3DSZp2Ok2jgw | 29368 |
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=kftFsIrCHjuwmLD4QAlK41lRffd5UOaT2bv933zB3zI | 6920 |
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=ryVkzeXo4YT5oVQNo0uoQD_95ENkcz7hlPiJVS_gz6g | 9262 |
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=PXaBLkqbyL83vUFd5fH-h1SELhyjJlYskx0JzmvXyUY | 6312 |
tests/_src/linear_haiku_test.py | sha256=Wn9evRP_uNrDR0PNbOd4jTyQPR4UORK5Mpnau1Eg6c4 | 6241 |
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.7.dist-info/LICENSE | sha256=yEtlBjogbzeC3Waw_DSwY2HE4RwHLdiF8K31yNF_fFw | 11346 |
e3nn_jax-0.20.7.dist-info/METADATA | sha256=w849P_M91yBbKNS8AuLcv8DkvpIGOSLR3_uqiIojRDw | 18604 |
e3nn_jax-0.20.7.dist-info/WHEEL | sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4 | 91 |
e3nn_jax-0.20.7.dist-info/top_level.txt | sha256=0m31RxjShR1n_tWu9pILzF_CaGGPddkhyL8loM7WoMk | 15 |
e3nn_jax-0.20.7.dist-info/RECORD | — | — |
top_level.txt
e3nn_jax
tests