Super helpful! We @onefact are a member of @uxlfoundation and need to benchmark jax as well - is this on your roadmap?

[jaanli]

Specifically, we need to benchmark these encoder-only models that I have built for hospitals: https://arxiv.org/abs/1904.05342 (under my maiden name). Happy to chat if of further interest! jaan@onefact.org.