Tarun Kumar Vangani

Thanks for the response. I solved the first issue in a rather hacky way and am sharing it here if it should help anyone, vterm for some reason does not...

Thanks for the prompt reply. Can you clarify what order.quantityPart means here: quantity = self.calc_quantity_from_part(order.symbol, order.quantityPart, order.price, order.side) is it the amount of coin bought by the master or fraction...

@rahulbatra85 thanks I did see that you had a fork for jax-triton. Anything semi broken I can try right now? Also let me know if I can take up something...

Thanks will do. Are there any binaries available. I am on LUMI and usually the step of converting from docker to singularity messes Jax containers for me.

@rahulbatra85 I couldn't get it work because our HPC has issues detecting GPUs inside singularity containers with JAX. Can you please share the exact branches/versions of the libs below so...

@rahulbatra85 thank you! I tried the setup proposed above and I run into this error on the line when I run MHA as shown at the start of the issue...

Further on making all the inputs bfloat16 I run into a different error: ``` >>> xq = jax.random.normal(rng, (bs, seqlen, n_heads, dim), dtype="bfloat16") >>> print(attention.mha(xq, xq, xq, None)) Traceback (most...

Bump when is this expected?

@rahulbatra85 thanks for the update. I will try and get back to you. On our main server (lumi) we are unfortunately stuck with rocm 5.6.1. Is there any way to...