samuela/torch2jax

Issues

• XlaRuntimeError: UNIMPLEMENTED: from_dlpack got array with non-default layout with minor-to-major dimensions (2,0,1), expected (2,1,0)

  #6 opened 6 months ago by truncs
  1

• Einops support

  #5 opened 6 months ago by jrichterpowell
  1

• Interoperability with Keops?

  #4 opened a year ago by adam-hartshorne
  6

• Alternative Torch2Jax

  #2 opened a year ago by adam-hartshorne
  2