clemisch/jaxtomo

Issues

• Performance regression with JAX 0.4.32 on CPU

  #3 opened 18 days ago by clemisch
  10

• FP slow, dominated by MemcpyD2H

  #2 opened 2 years ago by clemisch
  2

• jit around pmap'ed FP & BP is a problem when pmap'ing SIR cost functions

  #1 opened 2 years ago by clemisch
  0