jaxga
jaxga copied to clipboard
Published
20 hours ago •
RobinKa
RobinKa
Geometric Algebra package for JAX
Results
3
jaxga issues
Sort by
recently updated
recently updated
newest added
Thanks for making this library; I wanted to do something similar a few months ago but other things got into the way; awesome to see that the JAX ecosystem is...
- Previously was using a loop and add at index which gets unrolled, now using `segment_sum` to sum same output indices - 10x faster JIT on CPU, 6x faster JIT...
