Skip to content
Permalink

Comparing changes

This is a direct comparison between two commits made in this repository or its related repositories. View the default comparison for this range or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: ml-explore/mlx
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: c791d1eb90b325a0b6e722e88ef57400401273db
Choose a base ref
..
head repository: ml-explore/mlx
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: da591af8ff15e8685a2c54437a436c6265086eb3
Choose a head ref
Showing with 3 additions and 3 deletions.
  1. +3 −3 python/src/linalg.cpp
6 changes: 3 additions & 3 deletions python/src/linalg.cpp
Original file line number Diff line number Diff line change
@@ -199,7 +199,7 @@ void init_linalg(nb::module_& parent_module) {
in which case the default stream of the default device is used.
Returns:
tuple(array, array): The ``Q`` and ``R`` matrices.
tuple(array, array): ``Q`` and ``R`` matrices such that ``Q @ R = a``.
Example:
>>> A = mx.array([[2., 3.], [1., 2.]])
@@ -310,8 +310,8 @@ void init_linalg(nb::module_& parent_module) {
Returns:
array: If ``upper = False``, it returns a lower triangular ``L`` matrix such
that ``dot(L, L.T) = a``. If ``upper = True``, it returns an upper triangular
``U`` matrix such that ``dot(U.T, U) = a``.
that ``L @ L.T = a``. If ``upper = True``, it returns an upper triangular
``U`` matrix such that ``U.T @ U = a``.
)pbdoc");
m.def(
"cholesky_inv",