All pull requests

Pull requests

• Avoid use of deprecated xla_bridge.get_backend().live_buffers()

  #1809·copybara-service[bot] opened Feb 4, 2025Bot·

  ·

• Future-proof calls to jnp.solve on batched 1D inputs.

  #1806·copybara-service[bot] opened Jan 7, 2025Bot·

  ·

• Update references to JAX's GitHub repo

  #1805·copybara-service[bot] opened Dec 5, 2024Bot·

  ·

• Remove uses of deprecated jax.lax.tie_in

  #1795·copybara-service[bot] opened Jan 29, 2024Bot·

  ·

• Replace references to deprecated device_buffer attributes

  #1793·copybara-service[bot] opened Dec 6, 2023Bot·

  ·

• Refactoring to 1.5.1

  #1792·mmarcinmichal opened Nov 27, 2023·

  ·
  1

• Update core.py: fix docstring "y = Wx + b" -> "y = xW + b"

  #1784·arvyzukai opened Aug 29, 2023·

  ·
  1

• fix: write file typo

  #1782·0o001 opened Aug 19, 2023·

  ·

• Nick dory patch 1

  #1781·NickDory opened Aug 9, 2023·

  ·
  1

• Move the TF-NumPy einsum test from Trax to the TF-NumPy directory.

  #1777·copybara-service[bot] opened May 26, 2023Bot·

  ·

• Disables a failing remat test.

  #1763·copybara-service[bot] opened Nov 4, 2022Bot·

  ·

• Update inputs.py

  #1761·chrisrobertsoncypress opened Sep 26, 2022·

  ·
  1

• Fix issues related to new behavior of JAX DeviceArray.copy()

  #1734·copybara-service[bot] opened Apr 1, 2022Bot·

  ·

• tighten test_monte_carlo_bias_is_zero bound

  #1727·austereantelope opened Jan 25, 2022·

  ·

• Update trax.supervised.rst - This will now display documentation for all the classes defined inside the file

  #1726·vsnupoudel opened Jan 5, 2022·

  ·
  1

• ReformerLM api changes in text-gen notebook

  #1723·d0rc opened Dec 9, 2021·

  ·

• Add Pad layer with some Args

  cla: yes
  #1715·sunvod opened Nov 5, 2021Contributor·

  ·

• Update decoding.py

  cla: yes
  #1713·sunvod opened Oct 27, 2021Contributor·

  ·