2 ms·

Yes, that's right

• sroussey 3y ago
  Here? https://github.com/utkarsh530/GPUODEBenchmarks/blob/ef807198584e4634c82bfd6a82bb9241c51d56d1/GPU_ODE_JAX/bench_diffrax.py#L45 https://github.com/utkarsh530/GPUODEBenchmarks/blob/ef807198...
  • deleted 3y ago
    [deleted]