Supporting JAX about tensor-sensor HOT 8 CLOSED

I'm just about done with this feature request. Would you care to try it out? @ethanluoyc it is released so you can update if you like.

from tensor-sensor.

Heh, wow. jax looks really cool. I literally had not heard of this! Sure, I will poke around to see how this would work. parrt: check out https://sjmielke.com/jax-purify.htm

from tensor-sensor.

Thanks for the interest! It's getting more attention these days and I have found it to be quite useful in a broader sense of ML (probabilistic modeling). Please let me know if there is anything that I can help with.

from tensor-sensor.

Wow this is absolutely amazing! I will give it a try and let you know how it works!

from tensor-sensor.

Lemme know what you think. it even works for weird stuff:

jnp.fft.fft(x, norm="something weird")

from tensor-sensor.

I just tried it out. It really works like magic. My work involves working with lots of broadcasting and reshaping so when something goes wrong it was usually very hard to understand what the problem is. Your work has really made things a lot simpler!

from tensor-sensor.

That's great to hear, @ethanluoyc ! It would be very helpful if you could advertise TensorSensor in the JAX community :) thank you!

from tensor-sensor.

from tensor-sensor.