e3nn e3nn-jax Q A · Discussions · GitHub

FullyConnectedTensorProduct Feature Discrepancy in JAX vs. Torch #44
Q&A tai-dang11

Q&A Discussions

Ask the community for help

You must be logged in to vote

jit[just in time] compilation issue with radius_graph function

VinaySingh561 asked Jul 21, 2024 in Q&A · Unanswered

3
You must be logged in to vote

Discrepancy in output Irreps of e3nn.haiku.Linear with and without reinitializing IrrepsArray

yhwang17 asked Jul 21, 2024 in Q&A · Answered

2
You must be logged in to vote

Clarification: Mapping FullyConnectedTensorProduct and Gate from e3nn to e3nn-jax

nkzk-stan asked Mar 23, 2024 in Q&A · Unanswered

2
You must be logged in to vote

Query: Availability of JAX-based Implementations for e3nn.nn.models.gate_points_2101.Compose Module

VinaySingh561 asked Feb 22, 2024 in Q&A · Closed · Answered

1
You must be logged in to vote

Equivalent of FullyConnectedTensorProduct using Flax modules

Chronum94 asked Dec 11, 2023 in Q&A · Answered

7
You must be logged in to vote

FullyConnectedTensorProduct Feature Discrepancy in JAX vs. Torch #44

tai-dang11 asked Oct 1, 2023 in Q&A · Answered

4