Open koaning opened 4 years ago
I'm not sure what backend we'd want to use for this, but there may be something to be said to also allow for an implementation like this example.
It sounds like a fun exercise to build this in jax, but I don't know if the maintenance is going to be a pain. @MBrouns got strong opinions on this?
I'm not sure what backend we'd want to use for this, but there may be something to be said to also allow for an implementation like this example.
It sounds like a fun exercise to build this in jax, but I don't know if the maintenance is going to be a pain. @MBrouns got strong opinions on this?