Backend doc - Githubissues

JaxGaussianProcesses / GPJax

Gaussian processes in JAX.

https://docs.jaxgaussianprocesses.com/

Apache License 2.0

436 stars 51 forks source link

Backend doc #462

Closed thomaspinder closed 4 weeks ago

thomaspinder commented 1 month ago

Type of changes

[ ] Bug fix
[ ] New feature
[X] Documentation / docstrings
[ ] Tests
[ ] Other

Checklist

[X] I've formatted the new code by running poetry run pre-commit run --all-files --show-diff-on-failure before committing.
[-] I've added tests for new code.
[-] I've added docstrings for the new code.

Description

This PR introduces a notebook outlining the new NNX backend.

Issue Number: N/A