Skip to content

Adding AD rules for backends via ext #241

@kylebeggs

Description

@kylebeggs

Is this something the package would consider merging? As an exploratory task, I put together some rules for Mooncake and got a 10x speedup in the pullback. I used this as my test bed

function loss(points)
    tree = KDTree(points)
    _, dists = allknn(tree, 3)
    return -sum(sum, dists)
end

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions