Pinned Loading

1. trecs trecs Public

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

   Python

2. jax-mps jax-mps Public

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

   C++ 217 28

3. onnela-lab/gptools onnela-lab/gptools Public

   Gaussian processes on graphs and lattices in Stan.

   Python 19

4. numpyro numpyro Public

   Forked from pyro-ppl/numpyro

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

   Python

5. localscope localscope Public

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

   Python 22 3

6. flax-metrics flax-metrics Public

   Flax NNX implementation of common metrics.

   Python 1