search

gchq / coreax

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

Move score_matching.py and weights.py to equinox #656

Closed db091756 closed 2 weeks ago

db091756 commented 2 weeks ago

What's the new feature?

Classes in score_matching.py, weights.py should be ported to inheriting from eqx.Module.

What value does this add?

Consistency with rest of codebase, removal of tree flattening methods.

Is there an alternative you've considered?

No response

Additional context

No response