GitHub repository: smsharma/JaxKDTree
Primary language: Jupyter Notebook
Description: A Jax wrapper for cudaKDTree