search

jax-ml / coix

Inference Combinators in JAX
https://coix.readthedocs.io/en/latest/
Apache License 2.0
43 stars 2 forks source link

Oryx backend is not working as expected #27

Open fehiepsi opened 7 months ago

fehiepsi commented 7 months ago

Despite that the annealing example is working. Other examples like gmm and dmm do not give expected outputs.