[tensor-shapes] add JAX BERT example #1496
codspeed.yml
on: pull_request
Microbenchmarks (simulation)
6m 3s
PyTorch benchmarks (walltime)
0s