Commits

Breadcrumbs

History for

onmain

User selector

All users

All time

Commit History

Commits on Aug 19, 2024

• [shard_map docs]: Fix doc typos

  Show description for 3e764f6
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  3e764f6

  Copy full SHA for 3e764f6

• 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

Commits on Aug 16, 2024

• 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

• document jax.Array methods and attributes

  jakevdp

  committed
  e9d6fd3

  Copy full SHA for e9d6fd3

Commits on Aug 15, 2024

• Merge pull request #23039 from froystig:docs3

  Show description for 88ed579
  

  jax authors

  committed
  88ed579

  Copy full SHA for 88ed579

Commits on Aug 14, 2024

• Document methods of custom_jvp/custom_vjp

  jakevdp

  committed
  91f5512

  Copy full SHA for 91f5512

• Introduce hermetic CUDA in Google ML projects.

  Show description for 599c13a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  599c13a

  Copy full SHA for 599c13a

• docs: reorganize sections

  Show description for 12eebfe
  froystig

  committed
  12eebfe

  Copy full SHA for 12eebfe

• docs: fix `shard_map` guide headings

  Show description for d17edb4
  froystig

  committed
  d17edb4

  Copy full SHA for d17edb4

Commits on Aug 13, 2024

• Merge pull request #23036 from froystig:docs2

  Show description for d2b85a4
  

  jax authors

  committed
  d2b85a4

  Copy full SHA for d2b85a4

• docs: tidy up titles and headings

  Show description for 3c223cd
  froystig

  committed
  3c223cd

  Copy full SHA for 3c223cd

• Merge pull request #23027 from froystig:docs

  Show description for 23effba
  

  jax authors

  committed
  23effba

  Copy full SHA for 23effba

• docs: more sentence case

  froystig

  committed
  09e7311

  Copy full SHA for 09e7311

• Merge pull request #23023 from froystig:docs3

  Show description for 833560d
  

  jax authors

  committed
  833560d

  Copy full SHA for 833560d

• docs: sentence case page titles, section headings, some content

  froystig

  committed
  b8f8b7b

  Copy full SHA for b8f8b7b

Commits on Aug 12, 2024

• docs: sentence case index and sub-index headings

  Show description for 2644299
  froystig

  committed
  2644299

  Copy full SHA for 2644299

• Add a `jax.process_indices` function.

  Show description for 60bf5b7
  

  dfm

  authored and

  jax authors

  committed
  60bf5b7

  Copy full SHA for 60bf5b7

Commits on Aug 11, 2024

• Merge pull request #22991 from froystig:keys2

  Show description for d2fa884
  

  jax authors

  committed
  d2fa884

  Copy full SHA for d2fa884

• Merge pull request #22989 from froystig:keys

  Show description for 312ebdb
  

  jax authors

  committed
  312ebdb

  Copy full SHA for 312ebdb

• emphasize typed over legacy RNG keys in `random` module docs

  Show description for dd535d8
  froystig

  committed
  dd535d8

  Copy full SHA for dd535d8

• update README and several docs to typed RNG keys

  froystig

  committed
  371935c

  Copy full SHA for 371935c

• indent consistently in auto-parallelization and shard_map tutorials

  froystig

  committed
  ded5b53

  Copy full SHA for ded5b53

Commits on Aug 9, 2024

• Merge branch 'main' into pallas_distr_docs

  justinjfu

  authored
  bdb0330

  Copy full SHA for bdb0330

• [Pallas] Add pallas distributed computation tutorial

  justinjfu

  committed
  dcd186f

  Copy full SHA for dcd186f

Commits on Aug 8, 2024

• Remove PositionalSharding from distributed array doc

  yashk2810

  committed
  7f8a4c8

  Copy full SHA for 7f8a4c8

Commits on Aug 7, 2024

• [pallas] Add a warning message about experimental and incomplete status

  gnecula

  committed
  64eb8e9

  Copy full SHA for 64eb8e9

Commits on Aug 6, 2024

• CI: pin sphinx to avoid build errors on 8.0

  jakevdp

  committed
  4f8c5a3

  Copy full SHA for 4f8c5a3

Commits on Aug 5, 2024

• Merge pull request #22141 from dfm:update-cuda-call-example-to-ffi-call

  Show description for 44a8c98
  

  jax authors

  committed
  44a8c98

  Copy full SHA for 44a8c98

• [pallas] Enable check for GPU lowering that tensor sizes are power of 2

  Show description for 9b35b76
  

  gnecula

  authored and

  jax authors

  committed
  9b35b76

  Copy full SHA for 9b35b76

Commits on Aug 2, 2024

• Update CUDA custom call example code to use `ffi_call`.

  Show description for 5474e0e
  dfm

  committed
  5474e0e

  Copy full SHA for 5474e0e

Commits on Aug 1, 2024

• Merge pull request #22095 from dfm:ffi-call-tutorial

  Show description for aa9e1e4
  

  jax authors

  committed
  aa9e1e4

  Copy full SHA for aa9e1e4

• Add ffi_call tutorial

  Show description for 0b4800a
  dfm

  committed
  0b4800a

  Copy full SHA for 0b4800a

• [array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api

  jakevdp

  committed
  14fa062

  Copy full SHA for 14fa062

• [pallas] Add error message for block_shapes of rank less than 1.

  Show description for 43163ff
  

  gnecula

  authored and

  jax authors

  committed
  43163ff

  Copy full SHA for 43163ff

Commits on Jul 31, 2024

• Merge pull request #22746 from gnecula:pallas_consts

  Show description for d696813
  

  jax authors

  committed
  d696813

  Copy full SHA for d696813