• Changes
  
  • Support for Python 3.7 has been dropped, in accordance with JAX's
    {ref}version-support-policy.
    
  • The behavior of XLA_PYTHON_CLIENT_MEM_FRACTION=.XX has been changed to allocate XX% of
    the total GPU memory instead of the previous behavior of using currently available GPU memory
    to calculate preallocation. Please refer to
    GPU memory allocation for
    more details.
    
  • The deprecated method .block_host_until_ready() has been removed. Use
    .block_until_ready() instead.