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 16 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!

loss got nan.

Where is the normalisation assumption made and how does it affect the model if a different normalisation is used?

Complex support for expm1

Would appreciate it if I could get some pointers here about the failed test cases