Ramansh Sharma

icon indicating a website link https://rsmath.github.io/ icon indicating an email [email protected]

icon location Salt Lake City icon location PhD student at the Kahlert School of Computing at University of Utah. PINNs are cool.

Results 14 comments of Ramansh Sharma

not working with kitty terminal

Haha, I hope it becomes clear soon :)

not working with kitty terminal

Ahh, I see. I had that too, and then I removed it when I tried your suggestion. I'm quite glad vim-oscyank works with Kitty now. It'd be helpful to put...

not working with kitty terminal

Yes, I think so. Thank you again for your help.

hk.switch does not work inside a hk.vmap function when hk.set_state is used

Code ```python import haiku as hk import numpy as np import jax import jax.numpy as jnp def forward(input_): # hk.set_state("some_state_val", 2) # line that causes the problem def func(i, x):...