Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Aug 20, 2024

• Skip the global jit cpp cache if in/out_layouts are not None

  Show description for 1ab6279
  

  yashk2810

  authored and

  jax authors

  committed
  1ab6279

  Copy full SHA for 1ab6279

• Merge pull request #23130 from vfdev-5:nogil-cpu_feature_guard

  Show description for d148ade
  

  jax authors

  committed
  d148ade

  Copy full SHA for d148ade

• Added PyUnstable_Module_SetGIL to PyInit_cpu_feature_guard

  vfdev-5

  committed
  6546c48

  Copy full SHA for 6546c48

• Merge pull request #23129 from vfdev-5:add-no-gil-pybind-mlir-dialects

  Show description for b82bcdc
  

  jax authors

  committed
  b82bcdc

  Copy full SHA for b82bcdc

Commits on Aug 19, 2024

• If input layouts are specified via `in_shardings` to `jit` and the array that the jitted function is called with is uncommitted, reshard the input array to the layout specified by the user.

  Show description for 6e1c236
  

  yashk2810

  authored and

  jax authors

  committed
  6e1c236

  Copy full SHA for 6e1c236

• Added py::mod_gil_not_used() to PYBIND11_MODULE register_jax_dialects

  vfdev-5

  committed
  b1b3ea2

  Copy full SHA for b1b3ea2

• Update XLA dependency to use revision

  Show description for 292161a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  292161a

  Copy full SHA for 292161a

• [shard_map docs]: Fix doc typos

  Show description for 3e764f6
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  3e764f6

  Copy full SHA for 3e764f6

• Merge pull request #22523 from mattjj:scan-attrs-fix

  Show description for f1974b6
  

  jax authors

  committed
  f1974b6

  Copy full SHA for f1974b6

• fix basic scan bug with attrs

  mattjj

  committed
  ef82cb2

  Copy full SHA for ef82cb2

• Merge pull request #23102 from rajasekharporeddy:testbranch1

  Show description for ae211a1
  

  jax authors

  committed
  ae211a1

  Copy full SHA for ae211a1

• Merge pull request #22586 from cool-RR:logdir-pathlib

  Show description for 1563897
  

  jax authors

  committed
  1563897

  Copy full SHA for 1563897

• Normalize "interpreter mode" to "interpret mode", and "InterpreterTest" to "InterpretTest"

  Show description for 36739e8
  

  ayaka14732

  authored and

  jax authors

  committed
  36739e8

  Copy full SHA for 36739e8

• Merge pull request #23045 from selamw1:diag_docstring

  Show description for d115c1c
  

  jax authors

  committed
  d115c1c

  Copy full SHA for d115c1c

• Better docs for jnp.fmin and fmax

  rajasekharporeddy

  committed
  351b780

  Copy full SHA for 351b780

• Merge pull request #23094 from jakevdp:basearray-annotations

  Show description for 0666ccc
  

  jax authors

  committed
  0666ccc

  Copy full SHA for 0666ccc

• Merge pull request #23103 from rajasekharporeddy:testbranch2

  Show description for 772e042
  

  jax authors

  committed
  772e042

  Copy full SHA for 772e042

• Refactor FFI shape inference functions to include dimension check.

  Show description for 30d54ec
  

  dfm

  authored and

  jax authors

  committed
  30d54ec

  Copy full SHA for 30d54ec

• Fix bug in ffi_lowering where custom layouts were ignored.

  Show description for 79c222e
  

  dfm

  authored and

  jax authors

  committed
  79c222e

  Copy full SHA for 79c222e

• Rollback for: Implement initial vmap over pallas_call w/ ragged inputs (via jumbles)

  Show description for 66a3f87
  

  apaszke

  authored and

  jax authors

  committed
  66a3f87

  Copy full SHA for 66a3f87

• Add support for shape polymorphism in ffi_lowering and move lu_pivots_to_permutation lowering out of jaxlib.

  Show description for dad2f57
  

  dfm

  authored and

  jax authors

  committed
  dad2f57

  Copy full SHA for dad2f57

Commits on Aug 18, 2024

• Update XLA dependency to use revision

  Show description for 05792c9
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  05792c9

  Copy full SHA for 05792c9

• [numpy] Fix test failures under NumPy 2.0.

  Show description for ba5b081
  

  hawkinsp

  authored and

  jax authors

  committed
  ba5b081

  Copy full SHA for ba5b081

Commits on Aug 17, 2024

• Update XLA dependency to use revision

  Show description for b957f8b
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  b957f8b

  Copy full SHA for b957f8b

• Remove the unused _CONV_PREFERRED_ELEMENT_TYPE_DESCRIPTION

  rajasekharporeddy

  committed
  e39c868

  Copy full SHA for e39c868

• Improve type annotations for jax.Array methods

  jakevdp

  committed
  dd697a9

  Copy full SHA for dd697a9

• Add logging the jax2tf `mlir_module_serialized` module size.

  Show description for ce4dae1
  

  maxwillzq

  authored and

  selamw1

  committed
  ce4dae1

  Copy full SHA for ce4dae1

• Match the argument name with the name in `Args` section in docstring

  Show description for 0c543ae
  

  yashk2810

  authored and

  jax authors

  committed
  0c543ae

  Copy full SHA for 0c543ae

Commits on Aug 16, 2024

• Merge pull request #23088 from vfdev-5:update-build-build-py

  Show description for 3a6ff86
  

  jax authors

  committed
  3a6ff86

  Copy full SHA for 3a6ff86

• Update XLA dependency to use revision

  Show description for 8b831b8
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  8b831b8

  Copy full SHA for 8b831b8

• Minor fixes and documentation update for custom hermetic Python interpreter support.

  Show description for 60cc041
  

  vam-google

  authored and

  jax authors

  committed
  60cc041

  Copy full SHA for 60cc041

• Merge pull request #23086 from ROCm:ci_build_fix

  Show description for 3d942ef
  

  jax authors

  committed
  3d942ef

  Copy full SHA for 3d942ef

• Merge pull request #23080 from jakevdp:array-doc

  Show description for 1127f49
  

  jax authors

  committed
  1127f49

  Copy full SHA for 1127f49

• Implement initial vmap over pallas_call w/ ragged inputs (via jumbles)

  Show description for 24394a1
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  24394a1

  Copy full SHA for 24394a1

• document jax.Array methods and attributes

  jakevdp

  committed
  e9d6fd3

  Copy full SHA for e9d6fd3