EnzymeAD/Enzyme-JAX

Issues

• GPU CI of Jax not thinking its running on GPU

  #122 opened 23 days ago by wsmoses
  2

• Not working for 3.11

  #5 opened a month ago by martinjm97
  2

• Pip package needs updating to v0.0.5

  #13 opened a month ago by croci
  2

• Tracking issue for missing HLO derivatives

  #88 opened 4 months ago by mofeing
  0

• `test.py::EnzymePipeline` fails on `enzyme_call.optimize_module`

  #85 opened 3 months ago by mofeing
  1

• Missing differentiation rules for `einsum`, `unary_einsum`

  #82 opened 4 months ago by mofeing
  0

• HLO Canonicalizations Todo list

  #54 opened 6 months ago by wsmoses
  31

• pip does not automatically import jax

  #4 opened 9 months ago by martinjm97
  0

• Add support for `jacrev`, `jacfwd`, `hessian`, `vmap`

  #9 opened a year ago by martinjm97
  3