Sun-Xiaohui
Sun-Xiaohui
@jakevdp Thanks, that helps! However, I found that if I have a `Module` developed based on `rnn.lstm`, forward value between `model.apply()` and `jax.jit(model.apply)()` does not match, here is a minimal...
Hello, JAX team and Flax team, is fixing this bug in your latest work plan？
@icm-ai 你好，请问你能成功安装flash_attn吗，我在jetson orin nx上尝试多次都失败了，是否有预编译好的whl？
@chiamp Hello, Is it working as expected, or did I make a mistake? Thanks.