Thanks to visit codestin.com
Credit goes to github.com

Skip to content

FEA Array API support in GaussianProcessRegressor - #33096

Draft
betatim wants to merge 18 commits into
scikit-learn:mainfrom
betatim:array-api-gaussian-processs-regressor
Draft

FEA Array API support in GaussianProcessRegressor#33096
betatim wants to merge 18 commits into
scikit-learn:mainfrom
betatim:array-api-gaussian-processs-regressor

Conversation

@betatim

@betatim betatim commented Jan 16, 2026 •
edited
Loading

Copy link
Copy Markdown
Member

Reference Issues/PRs

AFAIK there isn't one.

What does this implement/fix? Explain your changes.

This rewrites the GaussianProcessRegressor to use the array API.

AI usage disclosure

I used AI assistance for:

  • Code generation (e.g., when writing an implementation or fixing a bug)
  • Test/benchmark generation
  • Documentation (including examples)
  • Research and understanding

Any other comments?

Todo

  • review the tests to condense them, diff is very large!

betatim added 5 commits January 8, 2026 11:54
@betatim
Convert GP kernels to array API
adb1989 
This mostly converts gaussian process kernels to be array API
compatible. Also adds new utils that are needed to make the conversion
possible.
@betatim
Convert GP regressor to array API
5f1b867
@betatim
Use solve_triangular for better performance
681d47c
@betatim
Improve performance of torch's fit()
84c93c0
@betatim
Optimise distance calculation
73ccd19
@github-actions github-actions Bot removed the CI:Linter failure The linter CI is failing on this PR label Jan 16, 2026
ogrisel
ogrisel reviewed Jan 19, 2026
View reviewed changes
Comment thread sklearn/gaussian_process/_gpr.py Outdated
@ogrisel ogrisel moved this to In Progress in Array API Jan 22, 2026
@github-actions github-actions Bot removed the CUDA CI label Jan 30, 2026
@betatim
Add cuvs to the CUDA CI
029459d 
cupy relies on cuvs to provide cupyx.scipy.spatial.distance.cdist
and friends. Relocking the environment also updates other
dependencies like pandas due to relocking. The cupy version changes
due to adding cuvs.
@github-actions github-actions Bot removed the CUDA CI label Jan 30, 2026
This was referenced Feb 2, 2026
Closed
Merged
@betatim betatim added the CUDA CI label Feb 5, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 5, 2026
@betatim betatim added the CUDA CI label Feb 5, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 5, 2026
@betatim
Use xp.matrix_transpose instead of .mT
d912c88 
The array attribute isn't available in all array libraries
@betatim betatim added the CUDA CI label Feb 5, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 5, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 11, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 11, 2026
@github-actions github-actions Bot removed the CUDA CI label Feb 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@ogrisel ogrisel ogrisel left review comments

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

Array API module:gaussian_process module:utils

Projects

Array API
Status: In Progress

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@betatim @ogrisel