Getting Started
jax.debug.print
jax.debug.breakpoint
checkify
Reference Documentation
Advanced JAX Tutorials
Notes
Developer documentation
API documentation
Make a copy of a matrix with elements above the kth diagonal zeroed.
LAX-backend implementation of scipy.linalg._special_matrices.tril().
scipy.linalg._special_matrices.tril()
Original docstring below.
m (array_like) – Matrix whose elements to return
k (int, optional) – Diagonal above which to zero elements. k == 0 is the main diagonal, k < 0 subdiagonal and k > 0 superdiagonal.
tril – Return is the same shape and type as m.
ndarray