Yi Heng Lim

icon indicating an email [email protected]

icon location London, England. Penang, Malaysia. icon location MEng. in Information Engineering from University of Cambridge, working on Machine Learning research.

Results 8 comments of Yi Heng Lim

vmap over indexing issue

Hi, I have the same issue but it only occurs when ``torch.autograd.set_detect_anomaly(True)`` is set. Is this an expected behaviour?

Linear algebra module to-do list

What will be the difference between lstsq in xitorch and torch.linalg.lstsq?

sample pool instead of parameter space

Hi, is the documentation for bandit available?

Reproducibility

Thanks, two quick follow-up questions -- how did you set the seeds for the results in the paper and is the data split the exact same as the NeuralRDE paper?

[`Generate`] Fix `gradient_checkpointing` and `use_cache` bug for generate-compatible models

Would like to take GPT-2!

Add Flax Whisper for audio classification

I have contributed to a few good first issues on HF, would like to take this to learn JAX if available!

Add Flax Whisper for audio classification

@Shubhamai please go ahead with the PR!

Add Flax Whisper for audio classification

@sanchit-gandhi I will be happy to contribute to some of them, would be great if you have any suggestions on any particular models!