Skip to main content

Browse the archive

Revision 7b182ec8791c97ba61bbe186e956e7bdd340a0f0 authored by Brian Patton on 11 March 2024, 16:05:25 UTC, committed by jax authors on 11 March 2024, 16:56:55 UTC 
Modify test to detect erroneous sorting of negative int32 as larger than positive int32.
 
PiperOrigin-RevId: 614684199
1 parent 71ec6e3
Raw File
jax.extend.linear_util.rst 
``jax.extend.linear_util`` module
=================================

.. automodule:: jax.extend.linear_util

.. autosummary::
  :toctree: _autosummary

  StoreException
  WrappedFun
  cache
  merge_linear_aux
  transformation
  transformation_with_aux
  wrap_init
back to top