4 ms·

I'd be very interested to see what JAX can do on this problem.

• dekhn 5y ago
  https://github.com/google/jax-md https://github.com/google/jax-md
  • deleted 5y ago
    [deleted]