LobPCG

Finds the k largest/smallest eigenvalues using LOBPCG.

Abstract Signature:

LobPCG(A: Tensor, k: int = 1, B: Tensor)

PyTorch

API: torch.lobpcg
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.scipy.sparse.linalg.lobpcg
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: scipy.sparse.linalg.lobpcg
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.scipy.sparse.linalg.lobpcg
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.scipy.sparse.linalg.lobpcg
Strategy: Direct Mapping
Official Docs ↗