search

ACEsuit / Polynomials4ML.jl

Polynomials for ML: fast evaluation, batching, differentiation
MIT License
12 stars 5 forks source link

Custom Sparse Prod Pool kernel for 3 Embeddings #77

Closed cortner closed 9 months ago

cortner commented 9 months ago

This provides a custom pooling kernel for 3 embeddings, since the generic kernel doesn't play nice for the UF_ACE implementation I'm working on.