Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 17, 2024

• Use constant memory to pass in TMA descriptors to the kernel

  Show description for 4ea73bf
  

  apaszke

  authored and

  jax authors

  committed
  4ea73bf

  Copy full SHA for 4ea73bf

• Fixed a ValueError when a Pallas GPU kernel closed over array constants

  Show description for f67f2e0
  

  superbobry

  authored and

  jax authors

  committed
  f67f2e0

  Copy full SHA for f67f2e0

• Rollback jax-ml#21888, because it breaks multiple internal users

  Show description for 4913fff
  

  superbobry

  authored and

  jax authors

  committed
  4913fff

  Copy full SHA for 4913fff

Commits on Jun 16, 2024

• Update XLA dependency to use revision

  Show description for 595a620
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  595a620

  Copy full SHA for 595a620

• Merge pull request jax-ml#21897 from matsen:patch-1

  Show description for ae0127d
  

  jax authors

  committed
  ae0127d

  Copy full SHA for ae0127d

Commits on Jun 15, 2024

• Update XLA dependency to use revision

  Show description for 546a3a6
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  546a3a6

  Copy full SHA for 546a3a6

• Merge pull request jax-ml#21880 from gnecula:doc_poly

  Show description for 993fa61
  

  jax authors

  committed
  993fa61

  Copy full SHA for 993fa61

• [shape_poly] Add documentation for shape polymorphism

  Show description for b58ff2b
  gnecula

  committed
  b58ff2b

  Copy full SHA for b58ff2b

• Fixing installation docs: needed quotes; markdown fix

  matsen

  authored
  b585ece

  Copy full SHA for b585ece

• Further reduce the run time of pmap_test

  Show description for 5019167
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  5019167

  Copy full SHA for 5019167

• Bump up tolerance for tests using _testQdwh.

  Show description for 4e748e1
  

  bchetioui

  authored and

  jax authors

  committed
  4e748e1

  Copy full SHA for 4e748e1

Commits on Jun 14, 2024

• Merge pull request jax-ml#21888 from jakevdp:lax-mul-bool

  Show description for 193591b
  

  jax authors

  committed
  193591b

  Copy full SHA for 193591b

• lax.mul: accept boolean inputs

  jakevdp

  committed
  4f7cd03

  Copy full SHA for 4f7cd03

• Update XLA dependency to use revision

  Show description for 895b490
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  895b490

  Copy full SHA for 895b490

• Merge pull request jax-ml#21890 from hawkinsp:nightly

  Show description for 9977504
  

  jax authors

  committed
  9977504

  Copy full SHA for 9977504

• Update nightly installation instructions to reflect new jax/jaxlib dependency structure.

  hawkinsp

  committed
  d5844b7

  Copy full SHA for d5844b7

• Merge pull request jax-ml#21856 from jakevdp:dtype-array

  Show description for e15e8b8
  

  jax authors

  committed
  e15e8b8

  Copy full SHA for e15e8b8

• Merge all nightly wheels into jax_nightly_releases.html since we have dropped monolithic jaxlib CUDA wheels

  Show description for 8f315c2
  

  yashk2810

  authored and

  jax authors

  committed
  8f315c2

  Copy full SHA for 8f315c2

• Deprecate passing of arrays in place of dtypes.

  jakevdp

  committed
  9d5932a

  Copy full SHA for 9d5932a

• Stop using deprecated device= argument to jax.jit in effects tests

  Show description for 56f6e74
  

  apaszke

  authored and

  jax authors

  committed
  56f6e74

  Copy full SHA for 56f6e74

• Merge pull request jax-ml#21879 from gnecula:exp_fix_deprecated

  Show description for c2b47e3
  

  jax authors

  committed
  c2b47e3

  Copy full SHA for c2b47e3

• [export] Fix a couple of missed deprecated fields.

  Show description for 3b5b20c
  gnecula

  committed
  3b5b20c

  Copy full SHA for 3b5b20c

• [Pallas] Don't actually vmap if we're vmapping over axis size 1

  Show description for e126560
  

  sharadmv

  authored and

  jax authors

  committed
  e126560

  Copy full SHA for e126560

• Re-land jax-ml#21847

  Show description for a92fa54
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  a92fa54

  Copy full SHA for a92fa54

• Reduce the matrix size in testPmapMapVmapCombinations to reduce the test run time.

  Show description for 06ec7d1
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  06ec7d1

  Copy full SHA for 06ec7d1

Commits on Jun 13, 2024

• Merge pull request jax-ml#21871 from jakevdp:fix-warnings

  Show description for b907242
  

  jax authors

  committed
  b907242

  Copy full SHA for b907242

• fix some additional warnings related to jax-ml#21834

  jakevdp

  committed
  3346527

  Copy full SHA for 3346527

• Merge pull request jax-ml#21863 from jakevdp:dep-tracer-hash

  Show description for 8f5f8df
  

  jax authors

  committed
  8f5f8df

  Copy full SHA for 8f5f8df

• Get rid of the is_hermitian argument for lax.qdwh. If it was known that H was also positive semi-definite, the polar decomposition would be I*H. But for indefinite H, the QDWH algorithm does not di…

  Show description for c839b26
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  c839b26

  Copy full SHA for c839b26

• Merge pull request jax-ml#21864 from superbobry:pallas

  Show description for c36bb59
  

  jax authors

  committed
  c36bb59

  Copy full SHA for c36bb59

• Reverts 5aedafc

  Show description for 0bcc81c
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  0bcc81c

  Copy full SHA for 0bcc81c

• Add trace to stages.Wrapped and add docs for it.

  Show description for 4ef33fa
  

  yashk2810

  authored and

  jax authors

  committed
  4ef33fa

  Copy full SHA for 4ef33fa

• Merge pull request jax-ml#20728 from rajasekharporeddy:test_branch2

  Show description for fa5a162
  

  jax authors

  committed
  fa5a162

  Copy full SHA for fa5a162

• Merge pull request jax-ml#21865 from hawkinsp:docs

  Show description for 46899cb
  

  jax authors

  committed
  46899cb

  Copy full SHA for 46899cb

• Update XLA dependency to use revision

  Show description for e25528d
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  e25528d

  Copy full SHA for e25528d

Pagination

Previous

Next