gchq / coreax

A library for coreset algorithms, written in Jax for fast execution and GPU support.
Apache License 2.0
21 stars 2 forks source link

Investigate the potential for reduced compile times in `metrics.py` #665

Open tc85324 opened 3 months ago

tc85324 commented 3 months ago

What's the new feature?

See the discussion here: https://github.com/gchq/coreax/pull/661#discussion_r1642508262

What value does this add?

May reduce compilation time and hence time to first result, particularly helpful for small problems.

Is there an alternative you've considered?

No response

Additional context

No response