Closed copybara-service[bot] closed 2 weeks ago
Move the geometry creation code to pure numpy
When we don't want to jit or take the gradient of this pure numpy is quicker than jax numpy (no compilation and possibly less memory copies etc)
This speeds up the building of the sim from 10s to 5s for the iterhybrid rampup example
Move the geometry creation code to pure numpy
When we don't want to jit or take the gradient of this pure numpy is quicker than jax numpy (no compilation and possibly less memory copies etc)
This speeds up the building of the sim from 10s to 5s for the iterhybrid rampup example