Minqi comments

Repositories
Issues
Comments

Results 3 comments of


                                            Minqi

[Q] Is there a way to change the legend for all plots?

+1 This would be an incredibly huge time saver!

vmap of cond's predicate results in select, leading to unexpected compute/memory use

Hi there! I was wondering what the JAX team's latest thinking is regarding the behavior of lax.cond when batched via vmap. I find myself often running into the design pattern...

[BUG] DeepSpeed Ulysses zero3 compatibility

Hi @samadejacobs, what would be needed to support Z3 with Ulysses in the HF client?