gpjax 0.13.2__tar.gz → 0.13.4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (166) hide show
  1. gpjax-0.13.4/.github/FUNDING.yml +1 -0
  2. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/build_docs.yml +1 -1
  3. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/release.yml +10 -10
  4. {gpjax-0.13.2 → gpjax-0.13.4}/.gitignore +5 -0
  5. {gpjax-0.13.2 → gpjax-0.13.4}/PKG-INFO +13 -13
  6. {gpjax-0.13.2 → gpjax-0.13.4}/README.md +10 -10
  7. {gpjax-0.13.2 → gpjax-0.13.4}/docs/GOVERNANCE.md +7 -9
  8. {gpjax-0.13.2 → gpjax-0.13.4}/docs/contributing.md +12 -12
  9. {gpjax-0.13.2 → gpjax-0.13.4}/docs/scripts/sharp_bits_figure.py +1 -1
  10. gpjax-0.13.4/examples/heteroscedastic_inference.py +389 -0
  11. {gpjax-0.13.2 → gpjax-0.13.4}/examples/regression.py +34 -24
  12. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/__init__.py +3 -3
  13. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/citation.py +13 -0
  14. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/dataset.py +1 -1
  15. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/fit.py +1 -1
  16. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/gps.py +273 -63
  17. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/__init__.py +1 -1
  18. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/base.py +2 -2
  19. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/__init__.py +1 -1
  20. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/base.py +1 -1
  21. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/constant_diagonal.py +1 -1
  22. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/dense.py +1 -1
  23. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/diagonal.py +1 -1
  24. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/eigen.py +1 -1
  25. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/non_euclidean/__init__.py +1 -1
  26. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/non_euclidean/graph.py +18 -6
  27. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/non_euclidean/utils.py +1 -1
  28. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/nonstationary/__init__.py +1 -1
  29. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/nonstationary/arccosine.py +1 -1
  30. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/nonstationary/linear.py +1 -1
  31. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/nonstationary/polynomial.py +1 -1
  32. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/__init__.py +1 -1
  33. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/base.py +1 -1
  34. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/matern12.py +1 -1
  35. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/matern32.py +1 -1
  36. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/matern52.py +1 -1
  37. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/periodic.py +1 -1
  38. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/powered_exponential.py +1 -1
  39. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/rational_quadratic.py +1 -1
  40. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/rbf.py +1 -1
  41. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/utils.py +1 -1
  42. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/stationary/white.py +1 -1
  43. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/likelihoods.py +234 -0
  44. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/mean_functions.py +2 -2
  45. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/objectives.py +56 -1
  46. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/parameters.py +8 -1
  47. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/scan.py +1 -1
  48. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/variational_families.py +129 -0
  49. {gpjax-0.13.2 → gpjax-0.13.4}/mkdocs.yml +14 -3
  50. {gpjax-0.13.2 → gpjax-0.13.4}/pyproject.toml +4 -3
  51. {gpjax-0.13.2 → gpjax-0.13.4}/static/CONTRIBUTING.md +12 -12
  52. {gpjax-0.13.2 → gpjax-0.13.4}/tests/conftest.py +7 -0
  53. {gpjax-0.13.2 → gpjax-0.13.4}/tests/integration_tests.py +9 -2
  54. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_citations.py +16 -0
  55. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_dataset.py +1 -1
  56. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_fit.py +1 -1
  57. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_gps.py +159 -13
  58. gpjax-0.13.4/tests/test_heteroscedastic.py +407 -0
  59. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/__init__.py +1 -1
  60. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_base.py +1 -1
  61. gpjax-0.13.4/tests/test_kernels/test_non_euclidean.py +115 -0
  62. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_nonstationary.py +1 -1
  63. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_stationary.py +1 -1
  64. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_utils.py +1 -1
  65. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_mean_functions.py +17 -2
  66. {gpjax-0.13.2 → gpjax-0.13.4}/uv.lock +23 -0
  67. gpjax-0.13.2/.github/workflows/pr_greeting.yml +0 -62
  68. gpjax-0.13.2/tests/test_kernels/test_non_euclidean.py +0 -50
  69. {gpjax-0.13.2 → gpjax-0.13.4}/.github/CODE_OF_CONDUCT.md +0 -0
  70. {gpjax-0.13.2 → gpjax-0.13.4}/.github/ISSUE_TEMPLATE/01_BUG_REPORT.md +0 -0
  71. {gpjax-0.13.2 → gpjax-0.13.4}/.github/ISSUE_TEMPLATE/02_FEATURE_REQUEST.md +0 -0
  72. {gpjax-0.13.2 → gpjax-0.13.4}/.github/ISSUE_TEMPLATE/03_CODEBASE_IMPROVEMENT.md +0 -0
  73. {gpjax-0.13.2 → gpjax-0.13.4}/.github/ISSUE_TEMPLATE/04_DOCS_IMPROVEMENT.md +0 -0
  74. {gpjax-0.13.2 → gpjax-0.13.4}/.github/ISSUE_TEMPLATE/config.yml +0 -0
  75. {gpjax-0.13.2 → gpjax-0.13.4}/.github/codecov.yml +0 -0
  76. {gpjax-0.13.2 → gpjax-0.13.4}/.github/commitlint.config.js +0 -0
  77. {gpjax-0.13.2 → gpjax-0.13.4}/.github/dependabot.yml +0 -0
  78. {gpjax-0.13.2 → gpjax-0.13.4}/.github/labeler.yml +0 -0
  79. {gpjax-0.13.2 → gpjax-0.13.4}/.github/labels.yml +0 -0
  80. {gpjax-0.13.2 → gpjax-0.13.4}/.github/pull_request_template.md +0 -0
  81. {gpjax-0.13.2 → gpjax-0.13.4}/.github/release-drafter.yml +0 -0
  82. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/auto-label.yml +0 -0
  83. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/commit-lint.yml +0 -0
  84. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/integration.yml +0 -0
  85. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/ruff.yml +0 -0
  86. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/security-analysis.yml +0 -0
  87. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/test_docs.yml +0 -0
  88. {gpjax-0.13.2 → gpjax-0.13.4}/.github/workflows/tests.yml +0 -0
  89. {gpjax-0.13.2 → gpjax-0.13.4}/CITATION.bib +0 -0
  90. {gpjax-0.13.2 → gpjax-0.13.4}/LICENSE.txt +0 -0
  91. {gpjax-0.13.2 → gpjax-0.13.4}/Makefile +0 -0
  92. {gpjax-0.13.2 → gpjax-0.13.4}/docs/CODE_OF_CONDUCT.md +0 -0
  93. {gpjax-0.13.2 → gpjax-0.13.4}/docs/design.md +0 -0
  94. {gpjax-0.13.2 → gpjax-0.13.4}/docs/index.md +0 -0
  95. {gpjax-0.13.2 → gpjax-0.13.4}/docs/index.rst +0 -0
  96. {gpjax-0.13.2 → gpjax-0.13.4}/docs/installation.md +0 -0
  97. {gpjax-0.13.2 → gpjax-0.13.4}/docs/javascripts/katex.js +0 -0
  98. {gpjax-0.13.2 → gpjax-0.13.4}/docs/refs.bib +0 -0
  99. {gpjax-0.13.2 → gpjax-0.13.4}/docs/scripts/gen_examples.py +0 -0
  100. {gpjax-0.13.2 → gpjax-0.13.4}/docs/scripts/gen_pages.py +0 -0
  101. {gpjax-0.13.2 → gpjax-0.13.4}/docs/scripts/notebook_converter.py +0 -0
  102. {gpjax-0.13.2 → gpjax-0.13.4}/docs/sharp_bits.md +0 -0
  103. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/GP.pdf +0 -0
  104. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/GP.svg +0 -0
  105. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/bijector_figure.svg +0 -0
  106. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/css/gpjax_theme.css +0 -0
  107. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/favicon.ico +0 -0
  108. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/gpjax.mplstyle +0 -0
  109. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/gpjax_logo.pdf +0 -0
  110. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/gpjax_logo.svg +0 -0
  111. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/jaxkern/lato.ttf +0 -0
  112. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/jaxkern/logo.png +0 -0
  113. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/jaxkern/logo.svg +0 -0
  114. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/jaxkern/main.py +0 -0
  115. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/step_size_figure.png +0 -0
  116. {gpjax-0.13.2 → gpjax-0.13.4}/docs/static/step_size_figure.svg +0 -0
  117. {gpjax-0.13.2 → gpjax-0.13.4}/docs/stylesheets/extra.css +0 -0
  118. {gpjax-0.13.2 → gpjax-0.13.4}/docs/stylesheets/permalinks.css +0 -0
  119. {gpjax-0.13.2 → gpjax-0.13.4}/examples/backend.py +0 -0
  120. {gpjax-0.13.2 → gpjax-0.13.4}/examples/barycentres/barycentre_gp.gif +0 -0
  121. {gpjax-0.13.2 → gpjax-0.13.4}/examples/barycentres.py +0 -0
  122. {gpjax-0.13.2 → gpjax-0.13.4}/examples/classification.py +0 -0
  123. {gpjax-0.13.2 → gpjax-0.13.4}/examples/collapsed_vi.py +0 -0
  124. {gpjax-0.13.2 → gpjax-0.13.4}/examples/constructing_new_kernels.py +0 -0
  125. {gpjax-0.13.2 → gpjax-0.13.4}/examples/data/max_tempeature_switzerland.csv +0 -0
  126. {gpjax-0.13.2 → gpjax-0.13.4}/examples/data/yacht_hydrodynamics.data +0 -0
  127. {gpjax-0.13.2 → gpjax-0.13.4}/examples/deep_kernels.py +0 -0
  128. {gpjax-0.13.2 → gpjax-0.13.4}/examples/gpjax.mplstyle +0 -0
  129. {gpjax-0.13.2 → gpjax-0.13.4}/examples/graph_kernels.py +0 -0
  130. {gpjax-0.13.2 → gpjax-0.13.4}/examples/intro_to_gps/decomposed_mll.png +0 -0
  131. {gpjax-0.13.2 → gpjax-0.13.4}/examples/intro_to_gps/generating_process.png +0 -0
  132. {gpjax-0.13.2 → gpjax-0.13.4}/examples/intro_to_gps.py +0 -0
  133. {gpjax-0.13.2 → gpjax-0.13.4}/examples/intro_to_kernels.py +0 -0
  134. {gpjax-0.13.2 → gpjax-0.13.4}/examples/likelihoods_guide.py +0 -0
  135. {gpjax-0.13.2 → gpjax-0.13.4}/examples/oceanmodelling.py +0 -0
  136. {gpjax-0.13.2 → gpjax-0.13.4}/examples/poisson.py +0 -0
  137. {gpjax-0.13.2 → gpjax-0.13.4}/examples/uncollapsed_vi.py +0 -0
  138. {gpjax-0.13.2 → gpjax-0.13.4}/examples/utils.py +0 -0
  139. {gpjax-0.13.2 → gpjax-0.13.4}/examples/yacht.py +0 -0
  140. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/distributions.py +0 -0
  141. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/integrators.py +0 -0
  142. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/approximations/__init__.py +0 -0
  143. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/approximations/rff.py +0 -0
  144. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/kernels/computations/basis_functions.py +0 -0
  145. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/linalg/__init__.py +0 -0
  146. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/linalg/operations.py +0 -0
  147. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/linalg/operators.py +0 -0
  148. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/linalg/utils.py +0 -0
  149. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/numpyro_extras.py +0 -0
  150. {gpjax-0.13.2 → gpjax-0.13.4}/gpjax/typing.py +0 -0
  151. {gpjax-0.13.2 → gpjax-0.13.4}/static/paper.bib +0 -0
  152. {gpjax-0.13.2 → gpjax-0.13.4}/static/paper.md +0 -0
  153. {gpjax-0.13.2 → gpjax-0.13.4}/static/paper.pdf +0 -0
  154. {gpjax-0.13.2 → gpjax-0.13.4}/tests/__init__.py +0 -0
  155. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_gaussian_distribution.py +0 -0
  156. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_imports.py +0 -0
  157. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_integrators.py +0 -0
  158. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_approximations.py +0 -0
  159. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_kernels/test_computation.py +0 -0
  160. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_likelihoods.py +0 -0
  161. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_linalg.py +0 -0
  162. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_markdown.py +0 -0
  163. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_numpyro_extras.py +0 -0
  164. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_objectives.py +0 -0
  165. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_parameters.py +0 -0
  166. {gpjax-0.13.2 → gpjax-0.13.4}/tests/test_variational_families.py +0 -0
@@ -0,0 +1 @@
1
+ github: [thomaspinder]
@@ -61,7 +61,7 @@ jobs:
61
61
  uv run mkdocs build
62
62
 
63
63
  - name: Deploy Page 🚀
64
- uses: JamesIves/github-pages-deploy-action@v4.7.3
64
+ uses: JamesIves/github-pages-deploy-action@v4.7.4
65
65
  with:
66
66
  branch: gh-pages
67
67
  folder: site
@@ -3,22 +3,22 @@ name: "Release Automation"
3
3
  on:
4
4
  push:
5
5
  tags:
6
- - 'v*'
6
+ - "v*"
7
7
  workflow_dispatch:
8
8
  inputs:
9
9
  version:
10
- description: 'Version to release (e.g., v0.6.0)'
10
+ description: "Version to release (e.g., v0.6.0)"
11
11
  required: true
12
12
  type: string
13
13
  draft:
14
- description: 'Create draft release'
14
+ description: "Create draft release"
15
15
  required: false
16
16
  default: false
17
17
  type: boolean
18
18
 
19
19
  permissions:
20
20
  contents: write
21
- id-token: write # For PyPI trusted publishing
21
+ id-token: write # For PyPI trusted publishing
22
22
 
23
23
  jobs:
24
24
  validate-release:
@@ -60,13 +60,13 @@ jobs:
60
60
  fail-fast: false
61
61
  matrix:
62
62
  os: [ubuntu-latest, macos-latest]
63
- python-version: ['3.11', '3.12', '3.13']
63
+ python-version: ["3.11", "3.12", "3.13"]
64
64
  exclude:
65
65
  # Reduce matrix size - test all Python versions on Ubuntu, subset on others
66
66
  - os: macos-latest
67
- python-version: '3.11'
67
+ python-version: "3.11"
68
68
  - os: macos-latest
69
- python-version: '3.12'
69
+ python-version: "3.12"
70
70
 
71
71
  runs-on: ${{ matrix.os }}
72
72
 
@@ -113,7 +113,7 @@ jobs:
113
113
  - name: Set up Python
114
114
  uses: actions/setup-python@v6
115
115
  with:
116
- python-version: '3.11'
116
+ python-version: "3.11"
117
117
 
118
118
  - name: Install uv
119
119
  uses: astral-sh/setup-uv@v7
@@ -151,7 +151,7 @@ jobs:
151
151
  - name: Set up Python
152
152
  uses: actions/setup-python@v6
153
153
  with:
154
- python-version: '3.11'
154
+ python-version: "3.11"
155
155
 
156
156
  - name: Install uv
157
157
  uses: astral-sh/setup-uv@v7
@@ -246,7 +246,7 @@ jobs:
246
246
  CHANGELOG="$CHANGELOG- Model validation results available in CI artifacts\n\n"
247
247
 
248
248
  CHANGELOG="$CHANGELOG### 🔍 What's Changed\n"
249
- CHANGELOG="$CHANGELOG- Full diff: https://github.com/JaxGaussianProcesses/GPJax/compare/$PREVIOUS_TAG...$CURRENT_TAG\n\n"
249
+ CHANGELOG="$CHANGELOG- Full diff: https://github.com/thomaspinder/GPJax/compare/$PREVIOUS_TAG...$CURRENT_TAG\n\n"
250
250
 
251
251
  # Escape for GitHub output
252
252
  EOF_MARKER=$(dd if=/dev/urandom bs=15 count=1 status=none | base64)
@@ -153,3 +153,8 @@ node_modules/
153
153
 
154
154
  docs/api
155
155
  docs/_examples
156
+ local_libs/
157
+ local_papers/
158
+ GEMINI.md
159
+ AGENTS.md
160
+ plans/
@@ -1,10 +1,10 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gpjax
3
- Version: 0.13.2
3
+ Version: 0.13.4
4
4
  Summary: Gaussian processes in JAX.
5
5
  Project-URL: Documentation, https://docs.jaxgaussianprocesses.com/
6
- Project-URL: Issues, https://github.com/JaxGaussianProcesses/GPJax/issues
7
- Project-URL: Source, https://github.com/JaxGaussianProcesses/GPJax
6
+ Project-URL: Issues, https://github.com/thomaspinder/GPJax/issues
7
+ Project-URL: Source, https://github.com/thomaspinder/GPJax
8
8
  Author-email: Thomas Pinder <tompinder@live.co.uk>
9
9
  License: MIT
10
10
  License-File: LICENSE.txt
@@ -72,11 +72,11 @@ Description-Content-Type: text/markdown
72
72
  <!-- <h1 align='center'>GPJax</h1>
73
73
  <h2 align='center'>Gaussian processes in Jax.</h2> -->
74
74
  <p align="center">
75
- <img width="700" height="300" src="https://raw.githubusercontent.com/JaxGaussianProcesses/GPJax/main/docs/static/gpjax_logo.svg" alt="GPJax's logo">
75
+ <img width="700" height="300" src="https://raw.githubusercontent.com/thomaspinder/GPJax/main/docs/static/gpjax_logo.svg" alt="GPJax's logo">
76
76
  </p>
77
77
 
78
- [![codecov](https://codecov.io/gh/JaxGaussianProcesses/GPJax/branch/master/graph/badge.svg?token=DM1DRDASU2)](https://codecov.io/gh/JaxGaussianProcesses/GPJax)
79
- [![CodeFactor](https://www.codefactor.io/repository/github/jaxgaussianprocesses/gpjax/badge)](https://www.codefactor.io/repository/github/jaxgaussianprocesses/gpjax)
78
+ [![codecov](https://codecov.io/gh/thomaspinder/GPJax/branch/master/graph/badge.svg?token=DM1DRDASU2)](https://codecov.io/gh/thomaspinder/GPJax)
79
+ [![CodeFactor](https://www.codefactor.io/repository/github/thomaspinder/GPJax/badge)](https://www.codefactor.io/repository/github/thomaspinder/GPJax)
80
80
  [![Netlify Status](https://api.netlify.com/api/v1/badges/d3950e6f-321f-4508-9e52-426b5dae2715/deploy-status)](https://app.netlify.com/sites/endearing-crepe-c2d5fe/deploys)
81
81
  [![PyPI version](https://badge.fury.io/py/GPJax.svg)](https://badge.fury.io/py/GPJax)
82
82
  [![Conda Version](https://img.shields.io/conda/vn/conda-forge/gpjax.svg)](https://anaconda.org/conda-forge/gpjax)
@@ -101,19 +101,19 @@ with GP models.
101
101
 
102
102
  We would be delighted to receive contributions from interested individuals and
103
103
  groups. To learn how you can get involved, please read our [guide for
104
- contributing](https://github.com/JaxGaussianProcesses/GPJax/blob/main/docs/contributing.md).
104
+ contributing](https://github.com/thomaspinder/GPJax/blob/main/docs/contributing.md).
105
105
  If you have any questions, we encourage you to [open an
106
- issue](https://github.com/JaxGaussianProcesses/GPJax/issues/new/choose). For
106
+ issue](https://github.com/thomaspinder/GPJax/issues/new/choose). For
107
107
  broader conversations, such as best GP fitting practices or questions about the
108
108
  mathematics of GPs, we invite you to [open a
109
- discussion](https://github.com/JaxGaussianProcesses/GPJax/discussions).
109
+ discussion](https://github.com/thomaspinder/GPJax/discussions).
110
110
 
111
111
  Another way you can contribute to GPJax is through [issue
112
112
  triaging](https://www.codetriage.com/what). This can include reproducing bug reports,
113
113
  asking for vital information such as version numbers and reproduction instructions, or
114
114
  identifying stale issues. If you would like to begin triaging issues, an easy way to get
115
115
  started is to
116
- [subscribe to GPJax on CodeTriage](https://www.codetriage.com/jaxgaussianprocesses/gpjax).
116
+ [subscribe to GPJax on CodeTriage](https://www.codetriage.com/thomaspinder/GPJax).
117
117
 
118
118
  As a contributor to GPJax, you are expected to abide by our [code of
119
119
  conduct](docs/CODE_OF_CONDUCT.md). If you feel that you have either experienced or
@@ -127,7 +127,7 @@ where we can discuss the development of GPJax and broader support for Gaussian
127
127
  process modelling.
128
128
 
129
129
  We appreciate all [the contributors to
130
- GPJax](https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors) who have helped to shape
130
+ GPJax](https://github.com/thomaspinder/GPJax/graphs/contributors) who have helped to shape
131
131
  GPJax into the package it is today.
132
132
 
133
133
  # Supported methods and interfaces
@@ -141,7 +141,7 @@ GPJax into the package it is today.
141
141
  > - [**Laplace Approximation**](https://docs.jaxgaussianprocesses.com/_examples/classification/#laplace-approximation)
142
142
  > - [**Inference on Non-Euclidean Spaces**](https://docs.jaxgaussianprocesses.com/_examples/constructing_new_kernels/#custom-kernel)
143
143
  > - [**Inference on Graphs**](https://docs.jaxgaussianprocesses.com/_examples/graph_kernels/)
144
- > - [**Pathwise Sampling**](https://docs.jaxgaussianprocesses.com/_examples/spatial/)
144
+ > - [**Heteroscedastic Inference**](https://docs.jaxgaussianprocesses.com/_examples/heteroscedastic_inference/)
145
145
  > - [**Learning Gaussian Process Barycentres**](https://docs.jaxgaussianprocesses.com/_examples/barycentres/)
146
146
  > - [**Deep Kernel Regression**](https://docs.jaxgaussianprocesses.com/_examples/deep_kernels/)
147
147
  > - [**Poisson Regression**](https://docs.jaxgaussianprocesses.com/_examples/poisson/)
@@ -215,7 +215,7 @@ conda install --channel conda-forge gpjax
215
215
  Clone a copy of the repository to your local machine and run the setup
216
216
  configuration in development mode.
217
217
  ```bash
218
- git clone https://github.com/JaxGaussianProcesses/GPJax.git
218
+ git clone https://github.com/thomaspinder/GPJax.git
219
219
  cd GPJax
220
220
  uv venv
221
221
  uv sync --extra dev
@@ -1,11 +1,11 @@
1
1
  <!-- <h1 align='center'>GPJax</h1>
2
2
  <h2 align='center'>Gaussian processes in Jax.</h2> -->
3
3
  <p align="center">
4
- <img width="700" height="300" src="https://raw.githubusercontent.com/JaxGaussianProcesses/GPJax/main/docs/static/gpjax_logo.svg" alt="GPJax's logo">
4
+ <img width="700" height="300" src="https://raw.githubusercontent.com/thomaspinder/GPJax/main/docs/static/gpjax_logo.svg" alt="GPJax's logo">
5
5
  </p>
6
6
 
7
- [![codecov](https://codecov.io/gh/JaxGaussianProcesses/GPJax/branch/master/graph/badge.svg?token=DM1DRDASU2)](https://codecov.io/gh/JaxGaussianProcesses/GPJax)
8
- [![CodeFactor](https://www.codefactor.io/repository/github/jaxgaussianprocesses/gpjax/badge)](https://www.codefactor.io/repository/github/jaxgaussianprocesses/gpjax)
7
+ [![codecov](https://codecov.io/gh/thomaspinder/GPJax/branch/master/graph/badge.svg?token=DM1DRDASU2)](https://codecov.io/gh/thomaspinder/GPJax)
8
+ [![CodeFactor](https://www.codefactor.io/repository/github/thomaspinder/GPJax/badge)](https://www.codefactor.io/repository/github/thomaspinder/GPJax)
9
9
  [![Netlify Status](https://api.netlify.com/api/v1/badges/d3950e6f-321f-4508-9e52-426b5dae2715/deploy-status)](https://app.netlify.com/sites/endearing-crepe-c2d5fe/deploys)
10
10
  [![PyPI version](https://badge.fury.io/py/GPJax.svg)](https://badge.fury.io/py/GPJax)
11
11
  [![Conda Version](https://img.shields.io/conda/vn/conda-forge/gpjax.svg)](https://anaconda.org/conda-forge/gpjax)
@@ -30,19 +30,19 @@ with GP models.
30
30
 
31
31
  We would be delighted to receive contributions from interested individuals and
32
32
  groups. To learn how you can get involved, please read our [guide for
33
- contributing](https://github.com/JaxGaussianProcesses/GPJax/blob/main/docs/contributing.md).
33
+ contributing](https://github.com/thomaspinder/GPJax/blob/main/docs/contributing.md).
34
34
  If you have any questions, we encourage you to [open an
35
- issue](https://github.com/JaxGaussianProcesses/GPJax/issues/new/choose). For
35
+ issue](https://github.com/thomaspinder/GPJax/issues/new/choose). For
36
36
  broader conversations, such as best GP fitting practices or questions about the
37
37
  mathematics of GPs, we invite you to [open a
38
- discussion](https://github.com/JaxGaussianProcesses/GPJax/discussions).
38
+ discussion](https://github.com/thomaspinder/GPJax/discussions).
39
39
 
40
40
  Another way you can contribute to GPJax is through [issue
41
41
  triaging](https://www.codetriage.com/what). This can include reproducing bug reports,
42
42
  asking for vital information such as version numbers and reproduction instructions, or
43
43
  identifying stale issues. If you would like to begin triaging issues, an easy way to get
44
44
  started is to
45
- [subscribe to GPJax on CodeTriage](https://www.codetriage.com/jaxgaussianprocesses/gpjax).
45
+ [subscribe to GPJax on CodeTriage](https://www.codetriage.com/thomaspinder/GPJax).
46
46
 
47
47
  As a contributor to GPJax, you are expected to abide by our [code of
48
48
  conduct](docs/CODE_OF_CONDUCT.md). If you feel that you have either experienced or
@@ -56,7 +56,7 @@ where we can discuss the development of GPJax and broader support for Gaussian
56
56
  process modelling.
57
57
 
58
58
  We appreciate all [the contributors to
59
- GPJax](https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors) who have helped to shape
59
+ GPJax](https://github.com/thomaspinder/GPJax/graphs/contributors) who have helped to shape
60
60
  GPJax into the package it is today.
61
61
 
62
62
  # Supported methods and interfaces
@@ -70,7 +70,7 @@ GPJax into the package it is today.
70
70
  > - [**Laplace Approximation**](https://docs.jaxgaussianprocesses.com/_examples/classification/#laplace-approximation)
71
71
  > - [**Inference on Non-Euclidean Spaces**](https://docs.jaxgaussianprocesses.com/_examples/constructing_new_kernels/#custom-kernel)
72
72
  > - [**Inference on Graphs**](https://docs.jaxgaussianprocesses.com/_examples/graph_kernels/)
73
- > - [**Pathwise Sampling**](https://docs.jaxgaussianprocesses.com/_examples/spatial/)
73
+ > - [**Heteroscedastic Inference**](https://docs.jaxgaussianprocesses.com/_examples/heteroscedastic_inference/)
74
74
  > - [**Learning Gaussian Process Barycentres**](https://docs.jaxgaussianprocesses.com/_examples/barycentres/)
75
75
  > - [**Deep Kernel Regression**](https://docs.jaxgaussianprocesses.com/_examples/deep_kernels/)
76
76
  > - [**Poisson Regression**](https://docs.jaxgaussianprocesses.com/_examples/poisson/)
@@ -144,7 +144,7 @@ conda install --channel conda-forge gpjax
144
144
  Clone a copy of the repository to your local machine and run the setup
145
145
  configuration in development mode.
146
146
  ```bash
147
- git clone https://github.com/JaxGaussianProcesses/GPJax.git
147
+ git clone https://github.com/thomaspinder/GPJax.git
148
148
  cd GPJax
149
149
  uv venv
150
150
  uv sync --extra dev
@@ -14,23 +14,21 @@ following the BDFL model of governance. We have since moved to the governance mo
14
14
  [Specialty
15
15
  Library](https://github.com/OpenTechStrategies/open-source-archetypes/blob/main/arch-specialty-library.ltx)
16
16
  and benefited from a community of
17
- [contributors](https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors). This
17
+ [contributors](https://github.com/thomaspinder/GPJax/graphs/contributors). This
18
18
  document outlines the governance structure for the current status.
19
19
 
20
20
  ## Roles
21
21
  * Contributors: Anyone who contributes to GPJAx is considered a contributor. This
22
22
  includes submitting code, filing issues, reviewing pull requests, and participating in
23
23
  discussions. They are listed under:
24
- * [https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors](https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors)
24
+ * [https://github.com/thomaspinder/GPJax/graphs/contributors](https://github.com/thomaspinder/GPJax/graphs/contributors)
25
25
  * Core contributors: Core contributors are contributors who have made significant
26
26
  contributions to the GPJax project, for example large modules or functionality.
27
- * GPJax gardeners: Gardeners are core contributors who are responsible for maintaining
28
- the project and making decisions about its future direction. GPJax gardeners have the
27
+ * GPJax gardeners: Gardeners are core contributor(s) who are responsible for maintaining
28
+ the project and making decisions about its future direction. A GPJax gardener(s) has the
29
29
  ability to merge pull requests into the GPJax repository. GPJax gardeners also take on
30
30
  administrative tasks such as website maintenance.
31
- * Currently [daniel-dodd@](https://github.com/daniel-dodd),
32
- [henrymoss@](https://github.com/henrymoss), [st--@](https://github.com/st--), and
33
- [thomaspinder@](https://github.com/thomaspinder) are the gardeners of GPJax.
31
+ * Currently [thomaspinder@](https://github.com/thomaspinder) is the gardener of GPJax.
34
32
 
35
33
  ## Responsibility
36
34
  We cannot hold anyone responsible really since we are all doing free work here, but some
@@ -67,12 +65,12 @@ the project for core contributors.
67
65
  Anyone is welcome to contribute to the GPJax project. Contributions can be made in the
68
66
  form of code, documentation, or other forms of support. To learn more about how to
69
67
  contribute, please see the [contributing
70
- guide](https://github.com/JaxGaussianProcesses/GPJax/blob/main/static/CONTRIBUTING.md).
68
+ guide](https://github.com/thomaspinder/GPJax/blob/main/static/CONTRIBUTING.md).
71
69
 
72
70
 
73
71
  ## Code of conduct
74
72
  All contributors to the GPJax project are expected to follow the project's [code of
75
- conduct](https://github.com/JaxGaussianProcesses/GPJax/blob/main/.github/CODE_OF_CONDUCT.md).
73
+ conduct](https://github.com/thomaspinder/GPJax/blob/main/.github/CODE_OF_CONDUCT.md).
76
74
  The code of conduct outlines the expected behavior of contributors and helps to ensure a
77
75
  welcoming and productive environment for all.
78
76
 
@@ -6,38 +6,38 @@ GPJax welcomes contributions from interested individuals or groups. There are
6
6
  many ways to contribute, including:
7
7
 
8
8
  - Answering questions on our [discussions
9
- page](https://github.com/JaxGaussianProcesses/GPJax/discussions).
10
- - Raising [issues](https://github.com/JaxGaussianProcesses/GPJax/issues) related to bugs
9
+ page](https://github.com/thomaspinder/GPJax/discussions).
10
+ - Raising [issues](https://github.com/thomaspinder/GPJax/issues) related to bugs
11
11
  or desired enhancements.
12
12
  - Contributing or improving the
13
- [docs](https://github.com/JaxGaussianProcesses/GPJax/tree/main/docs) or
14
- [examples](https://github.com/JaxGaussianProcesses/GPJax/tree/master/docs/nbs).
15
- - Fixing outstanding [issues](https://github.com/JaxGaussianProcesses/GPJax/issues)
13
+ [docs](https://github.com/thomaspinder/GPJax/tree/main/docs) or
14
+ [examples](https://github.com/thomaspinder/GPJax/tree/master/docs/nbs).
15
+ - Fixing outstanding [issues](https://github.com/thomaspinder/GPJax/issues)
16
16
  (bugs).
17
- - Extending or improving our [codebase](https://github.com/JaxGaussianProcesses/GPJax).
17
+ - Extending or improving our [codebase](https://github.com/thomaspinder/GPJax).
18
18
 
19
19
 
20
20
  ## Code of conduct
21
21
 
22
22
  As a contributor to GPJax, you can help us keep the community open and
23
23
  inclusive. Please read and follow our [Code of
24
- Conduct](https://github.com/JaxGaussianProcesses/GPJax/blob/master/.github/CODE_OF_CONDUCT.md).
24
+ Conduct](https://github.com/thomaspinder/GPJax/blob/master/.github/CODE_OF_CONDUCT.md).
25
25
 
26
26
  ## Opening issues and getting support
27
27
 
28
28
  Please open issues on [Github Issue
29
- Tracker](https://github.com/JaxGaussianProcesses/GPJax/issues/new/choose). Here you can
29
+ Tracker](https://github.com/thomaspinder/GPJax/issues/new/choose). Here you can
30
30
  mention
31
31
 
32
32
  You can ask a question or start a discussion in the [Discussion
33
- section](https://github.com/JaxGaussianProcesses/GPJax/discussions) on Github.
33
+ section](https://github.com/thomaspinder/GPJax/discussions) on Github.
34
34
 
35
35
  ## Contributing to the source code
36
36
 
37
37
  Submitting code contributions to GPJax is done via a [GitHub pull
38
38
  request](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/proposing-changes-to-your-work-with-pull-requests/about-pull-requests).
39
39
  Our preferred workflow is to first fork the [GitHub
40
- repository](https://github.com/JaxGaussianProcesses/GPJax), clone it to your local
40
+ repository](https://github.com/thomaspinder/GPJax), clone it to your local
41
41
  machine, and develop on a _feature branch_. Once you're happy with your changes,
42
42
  install our `pre-commit hooks`, `commit` and `push` your code.
43
43
 
@@ -51,9 +51,9 @@ you through every detail!
51
51
 
52
52
  ### Step-by-step guide:
53
53
 
54
- 1. Click [here](https://github.com/JaxGaussianProcesses/GPJax/fork) to Fork GPJax's
54
+ 1. Click [here](https://github.com/thomaspinder/GPJax/fork) to Fork GPJax's
55
55
  codebase (alternatively, click the 'Fork' button towards the top right of
56
- the [main repository page](https://github.com/JaxGaussianProcesses/GPJax)). This
56
+ the [main repository page](https://github.com/thomaspinder/GPJax)). This
57
57
  adds a copy of the codebase to your GitHub user account.
58
58
 
59
59
  2. Clone your GPJax fork from your GitHub account to your local disk, and add
@@ -21,7 +21,7 @@ import matplotlib as mpl
21
21
  from matplotlib import patches
22
22
 
23
23
  plt.style.use(
24
- "https://raw.githubusercontent.com/JaxGaussianProcesses/GPJax/main/docs/examples/gpjax.mplstyle"
24
+ "https://raw.githubusercontent.com/thomaspinder/GPJax/refs/heads/main/examples/gpjax.mplstyle"
25
25
  )
26
26
  cols = mpl.rcParams["axes.prop_cycle"].by_key()["color"]
27
27