jeometric. Graph neural networks in JAX.
speculative_mamba. Python
SUPA. Python
pytorch_approximate_retrieval. Python
