Fix key splitting issues in api

jax-ml / coix

Inference Combinators in JAX

https://coix.readthedocs.io/en/latest/

Apache License 2.0

43 stars 2 forks source link

Closed fehiepsi closed 1 year ago

fehiepsi commented 1 year ago

In addition, fix a bug in oryx backend that introduces symbolic_zeros keyword to custom vjp call.