Pinned Loading

1. trecstrecsPublic

   Transformer-based playlist completion model trained on the Million Playlist Dataset.

   Python

2. jax-mpsjax-mpsPublic

   A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.

   C++ 209 26

3. onnela-lab/gptoolsonnela-lab/gptoolsPublic

   Gaussian processes on graphs and lattices in Stan.

   Python 19

4. numpyronumpyroPublic

   Forked from pyro-ppl/numpyro

   Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

   Python

5. localscopelocalscopePublic

   Restrict the scope of functions for reproducible code execution and peace of mind.

   Python 21 3

6. flax-metricsflax-metricsPublic

   Flax NNX implementation of common metrics.

   Python 1