robertodr
commented
3 years ago - Profile execution: https://jax.readthedocs.io/en/latest/profiling.html Not sure whether the tools at this link are only relevant on GPU/TPU.
- Use JIT
- For higher orders, only generate unique index combos and then replicate to full format using symmetry.