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 3 months ago by truncs
  1

• Einops support

  #5 opened 4 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