<commit_before>
jax.numpy package
=================

.. currentmodule:: jax.numpy

.. automodule:: jax.numpy

.. autosummary::
  :toctree: _autosummary

    abs
    absolute
    add
    all
    allclose
    alltrue
    amax
    amin
    angle
    any
    append
    arange
    arccos
    arccosh
    arcsin
    arcsinh
    arctan
    arctan2
    arctanh
    argmax
    argmin
    argsort
    around
    array
    asarray
    atleast_1d
    atleast_2d
    bitwise_and
    bitwise_not
    bitwise_or
    bitwise_xor
    broadcast_arrays
    broadcast_to
    ceil
    clip
    column_stack
    concatenate
    conj
    conjugate
    cos
    cosh
    count_nonzero
    cumsum
    cumprod
    cumproduct
    diag
    diag_indices
    diagonal
    divide
    divmod
    dot
    einsum
    equal
    exp
    exp2
    expand_dims
    expm1
    eye
    fabs
    flip
    fliplr
    flipud
    floor
    floor_divide
    fmod
    full
    full_like
    geomspace
    greater
    greater_equal
    hstack
    identity
    imag
    inner
    isclose
    isfinite
    isinf
    isnan
    isneginf
    isposinf
    kron
    left_shift
    less
    less_equal
    linspace
    log
    log10
    log1p
    log2
    logaddexp
    logaddexp2
    logical_and
    logical_not
    logical_or
    logical_xor
    logspace
    matmul
    max
    maximum
    mean
    meshgrid
    min
    minimum
    mod
    moveaxis
    multiply
    nan_to_num
    nancumprod
    nancumsum
    nanmax
    nanmin
    nanprod
    nansum
    negative
    not_equal
    ones
    ones_like
    outer
    pad
    polyval
    power
    prod
    ravel
    real
    remainder
    repeat
    reshape
    right_shift
    rot90
    round
    row_stack
    sign
    sin
    sinh
    sometrue
    sort
    split
    sqrt
    square
    squeeze
    stack
    std
    subtract
    sum
    swapaxes
    take
    take_along_axis
    tan
    tanh
    tensordot
    trace
    transpose
    tri
    tril
    triu
    true_divide
    var
    vdot
    vstack
    where
    zeros
    zeros_like

jax.numpy.linalg
----------------

.. automodule:: jax.numpy.linalg

.. autosummary::
  :toctree: _autosummary

  cholesky
  det
  eigh
  inv
  qr
  slogdet
  solve
  svd<commit_msg>Add new numpy functions to docs.<commit_after>
jax.numpy package
=================

.. currentmodule:: jax.numpy

.. automodule:: jax.numpy

.. autosummary::
  :toctree: _autosummary

    abs
    absolute
    add
    all
    allclose
    alltrue
    amax
    amin
    angle
    any
    append
    arange
    arccos
    arccosh
    arcsin
    arcsinh
    arctan
    arctan2
    arctanh
    argmax
    argmin
    argsort
    around
    array
    array_repr
    array_str
    asarray
    atleast_1d
    atleast_2d
    bitwise_and
    bitwise_not
    bitwise_or
    bitwise_xor
    broadcast_arrays
    broadcast_to
    ceil
    clip
    column_stack
    concatenate
    conj
    conjugate
    cos
    cosh
    count_nonzero
    cumsum
    cumprod
    cumproduct
    deg2rad
    degrees
    diag
    diag_indices
    diagonal
    divide
    divmod
    dot
    dsplit
    einsum
    equal
    exp
    exp2
    expand_dims
    expm1
    eye
    fabs
    flip
    fliplr
    flipud
    floor
    floor_divide
    fmod
    full
    full_like
    geomspace
    greater
    greater_equal
    hsplit
    hstack
    hypot
    identity
    imag
    inner
    isclose
    isfinite
    isinf
    isnan
    isneginf
    isposinf
    issubdtype
    issubsctype
    kron
    left_shift
    less
    less_equal
    linspace
    log
    log10
    log1p
    log2
    logaddexp
    logaddexp2
    logical_and
    logical_not
    logical_or
    logical_xor
    logspace
    matmul
    max
    maximum
    mean
    meshgrid
    min
    minimum
    mod
    moveaxis
    multiply
    nan_to_num
    nancumprod
    nancumsum
    nanmax
    nanmin
    nanprod
    nansum
    negative
    not_equal
    ones
    ones_like
    outer
    pad
    polyval
    power
    prod
    product
    rad2deg
    radians
    ravel
    real
    reciprocal
    remainder
    repeat
    reshape
    right_shift
    rot90
    round
    row_stack
    sign
    sin
    sinh
    sometrue
    sort
    split
    sqrt
    square
    squeeze
    stack
    std
    subtract
    sum
    swapaxes
    take
    take_along_axis
    tan
    tanh
    tensordot
    trace
    transpose
    tri
    tril
    triu
    true_divide
    var
    vdot
    vsplit
    vstack
    where
    zeros
    zeros_like

jax.numpy.linalg
----------------

.. automodule:: jax.numpy.linalg

.. autosummary::
  :toctree: _autosummary

  cholesky
  det
  eigh
  inv
  qr
  slogdet
  solve
  svd
{'fraction_non_alphanumeric': 0.02917657229306246, 'fraction_numerical': 0.0056191917008861034, 'mean_word_length': 1.463012240553486, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6842487', 'n_tokens_mistral': 1713, 'n_tokens_neox': 1686, 'n_words': 390}