Jed Homer comments

Results 8 comments of


                                            Jed Homer

Extension to 3D Wavelet Transform?

Thanks for your reply and the recommendations! The main thing I am looking to do is a Haar 3D-DWT as part of a parallelised layer in `Pytorch`. I think all...

Extension to 3D Wavelet Transform?

No I didn't - it's an area of mystery to me but I would be very interested in a discussion / looking into a solution! There doesn't seem to be...

How to learn y condition?

Following @rosinality's advice, it works with label conditioning on MNIST. I added a label conditioning to the affine coupling layers instead of the prior (this is good because the net...

Analog to `flax.struct.dataclass`

Thanks @patrick-kidger, I will definitely use `equinox` next time around, its design is incredible.

Question: States, `vmap`-construction-of-layers and `lax.scan`'ing.

Hi Johanna! Thank you for the reply! I might be missing something here... If I change the batch size e.g. `x = jnp.ones((7, 4))` the error returns. It seems like...

Question: States, `vmap`-construction-of-layers and `lax.scan`'ing.

@lockwo - thank you for this insight! This was what I was trying just now (didn't work out the tree mapping, been confused by tracers and what not...). If it's...

Question: States, `vmap`-construction-of-layers and `lax.scan`'ing.

Awesome stuff, I was hoping you'd see this @patrick-kidger :). Yeah I tried the param `xs` arg for `lax.scan` but I must have had a hidden bug or something in...

`filter_shard` test crashes on JAX 0.4.29

Ah this is interesting. I will have a look at this but not sure a priori!