bilevel_coresets

bilevel_coresets copied to clipboard

How to speed up the loop in build_with_representer_proxy_batch

1

Hi, thank you very much for providing the code! I've installed jax with CUDA, so now solving the kernel_fn is faster. However, the build_with_representer_proxy_batch is still quite slow, I assume...

pipilurj