cuequivariance_jax/SKILL.md,sha256=VBmZaKw8aes4VRdxqmDmaX0Hq0eteWxHppilEulb2GY,15420
cuequivariance_jax/VERSION,sha256=3CT-tb01CE2K4ypOr77BI1JwfUZiQB_LzJu9aWzed6k,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=k_h1oZjNBzHTAqTTiR5sz-Cf7YNGqTmY_ryQxo3NAZ8,8703
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.10.0.dist-info/METADATA,sha256=Lg7hI5GprUm-c9o7l4d2BKg99-tgtYpc0UdE7tD77GQ,15282
cuequivariance_jax-0.10.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
cuequivariance_jax-0.10.0.dist-info/licenses/LICENSE,sha256=jPPjFJOcJIq3OusQT0swtpFxTHHnAC85bMD1NWKODko,11361
cuequivariance_jax-0.10.0.dist-info/RECORD,,
