GPJax provides a low-level interface for Gaussian process models in JAX, structured to give researchers maximum flexibility in extending the code to suit their needs. The library is designed to make the implementation closely mirror the mathematics of GP models. It serves as a probabilistic programming tool for Bayesian inference using Gaussian processes.