cuequivariance_jax/SKILL.md,sha256=Auegs1YVrgZPIb5m8jp-6LE7za_p_7DkMo8mIulL86g,14384
cuequivariance_jax/VERSION,sha256=eV1rx5V00q7AOtnP7EBLuVCDyd0hDmUh4NQZl3LSjUQ,7
cuequivariance_jax/__init__.py,sha256=oM34kMqUoNtcjH4qNTbZTqIlkhocrw7WN1HgD8Ufgmg,2295
cuequivariance_jax/__main__.py,sha256=YLSMvOmi0BJ7baomkIGc-NG_JPFT9iNlw74jHlshKF8,1011
cuequivariance_jax/activation.py,sha256=D9Hwp3Xl03yvkegcXtJDxvqjJFkQrnEFABZkKuWtOU4,4798
cuequivariance_jax/benchmarking.py,sha256=XYQGJksrUEk8K1AGAzz99yly4mpzDph9CvCESlpn4jg,7577
cuequivariance_jax/equivariant_polynomial.py,sha256=QZ-rM8LjLg4YKKC9MKpF9HvLzi4lCbF5w5BRm6dsSgU,7892
cuequivariance_jax/ir_dict.py,sha256=_UfWDSQhcjNpWUoLfg60lvmqq4G1aCX07VuNj35uUB8,13093
cuequivariance_jax/nnx.py,sha256=c0C9e1myxnugtQtoqoB5GIiVBS5-ah7TkVNDs0qHGvY,8587
cuequivariance_jax/spherical_harmonics.py,sha256=_rcvuVG46ku1zuPsfg81TNP41uQ1HR1SV-CHZTvdA_U,4678
cuequivariance_jax/experimental/__init__.py,sha256=QAkt471FNZ3YuN-wRFNp5mCuuy9uUf-x8lcAz0GyCB4,811
cuequivariance_jax/experimental/indexed_linear.py,sha256=SqXKIJZr1HbXALhHgCrgepdYo8GFxLxJh7-TaF5tFtM,1054
cuequivariance_jax/experimental/utils.py,sha256=fgnuy2EmcUF8W8YHVLuozsbsB2WYPVQ-2yLDwv5Ho3I,4085
cuequivariance_jax/flax_linen/__init__.py,sha256=xAe4nRWtOZhZWyoj5i-DcE810ire5zPoHqFFBuEeWAA,751
cuequivariance_jax/flax_linen/layer_norm.py,sha256=Segx9E2hWl6ShY6SVYzumRe3hNko5r4k77WjxwdfwdY,1854
cuequivariance_jax/rep_array/rep_array_.py,sha256=riokyQ3n0eLlistj1TCIhkGGWvdhAPFprfLoa579_q4,22066
cuequivariance_jax/rep_array/rep_array_utils.py,sha256=-j10nWIsKnlSXaROnWFOE_Oj8dM2udhpDU4F4ZvOtqY,5844
cuequivariance_jax/rep_array/vmap.py,sha256=DZQISjKsZ2_DVbKaP4JNxPTUNYuQuT3saqZ5JrI8HNA,4016
cuequivariance_jax/segmented_polynomials/indexing_mode.py,sha256=cwf7V8Tl3R0GlEtH2Bu3_CPRg80w7M3vdpzLyiODCQM,986
cuequivariance_jax/segmented_polynomials/segmented_polynomial.py,sha256=3WZmnyVveNicbMD8I14uP1b7cz0ibEcpkkMdrWHfz74,30449
cuequivariance_jax/segmented_polynomials/segmented_polynomial_indexed_linear.py,sha256=62nATy9ZWe6VbyQpEYzlfnZ4ih-JiT89UblOPHfgDXk,14331
cuequivariance_jax/segmented_polynomials/segmented_polynomial_naive.py,sha256=DpMDFIjiOVi2ya3I7g3uMSzzJGFHttj9hQ1qqe1KU7M,14354
cuequivariance_jax/segmented_polynomials/segmented_polynomial_uniform_1d.py,sha256=9tJd0VcaPMALobv3lHsBl8HTCRbDjkIPohW3DnyIaac,7763
cuequivariance_jax/segmented_polynomials/utils.py,sha256=fPsUGDv6YMD_Tfx2B8KGY3MUQQApab_SAAhrdspHIGU,3513
cuequivariance_jax/triangle/__init__.py,sha256=7aBq2QT0Rl9IFzc7eBFSBrbCUcplBosfav220sFUvL8,942
cuequivariance_jax/triangle/_layer_norm_transpose.py,sha256=F6zg6lY65oasz4z1gX2UdudNhmIeAEKQFqXGvvgxeG8,19955
cuequivariance_jax/triangle/_naive_batching.py,sha256=wESjmxXe_1J2Mo7iWV98Z22AXG4RE6SmMK9dLMRMQsU,3681
cuequivariance_jax/triangle/_sigmoid_gated_dual_gemm.py,sha256=GKyNYA5K9Rt6PBi0ObJcBjiEbKlXiWuqaWdNdDxepfk,38593
cuequivariance_jax/triangle/_triangle_attention.py,sha256=lf3dIDLP_UJDbM47qPQiOX2TVsHeE-M9gM8G0iRgesQ,10182
cuequivariance_jax/triangle/_triangle_multiplicative_update.py,sha256=fGxMGsprkNfmGM8fI-n-VwV0nUeV7_3Rmt3H2nyElOA,17500
cuequivariance_jax/triangle/_utils.py,sha256=rkO0P1fqscGY_h_CVAXaW4QSDFXhwl_fFR5AFeMLBOU,1386
cuequivariance_jax/triangle/triton_utils.py,sha256=Hvk9u965guWgszRqz3NTo9V_VjcEShK8_Tkq4vKKuT0,12080
Third_party_attr.txt,sha256=D-9LjRNeyQHvBrae0BOCwJaW1Dzx747ZS6HU9lUSQtk,760502
cuequivariance_jax-0.11.0.dist-info/METADATA,sha256=BvsQG1a2Y_mhasDGHC-gyJxaiAcBFC0EMK0u1aUzhdg,15282
cuequivariance_jax-0.11.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
cuequivariance_jax-0.11.0.dist-info/licenses/LICENSE,sha256=jPPjFJOcJIq3OusQT0swtpFxTHHnAC85bMD1NWKODko,11361
cuequivariance_jax-0.11.0.dist-info/RECORD,,
