Commits

Branch selector

 jax-v0.3.16-ipu

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 19, 2023

• Switching IPU testing to Poplar SDK 3.2 and 3.3. (#25)

  balancap

  authored
  a43c1b3

  Copy full SHA for a43c1b3

Commits on Sep 16, 2023

• Improve IPU PjRt client XLA non-standard layout handling. (#22)

  Show description for fcdff26
  balancap

  authored
  fcdff26

  Copy full SHA for fcdff26

Commits on Sep 13, 2023

• Add donated buffer multi-functions edge case unit test. (#21)

  Show description for be67565
  balancap

  authored
  be67565

  Copy full SHA for be67565

Commits on Sep 11, 2023

• Basic support `jax.debug.callback` in JAX IPU experimental. (#18)

  Show description for e23130d
  balancap

  authored
  e23130d

  Copy full SHA for e23130d

Commits on Jul 14, 2023

• Implement IPU threefry algorithm custom vertex. (#16)

  Show description for 8fedd0b
  balancap

  authored
  8fedd0b

  Copy full SHA for 8fedd0b

Commits on Jul 6, 2023

• Quickstart updates (#14)

  Show description for 527dc50
  awf

  authored
  527dc50

  Copy full SHA for 527dc50

Commits on Apr 28, 2023

• Support simple C++ interface for building custom JAX primitives on IPU. (#25)

  Show description for 003bcbd
  balancap

  authored
  003bcbd

  Copy full SHA for 003bcbd

• JAX IPU CI testing multiple Poplar SDKs. (#11)

  Show description for f81de22
  balancap

  authored
  f81de22

  Copy full SHA for f81de22

• JAX IPU public CI generating Python wheels HTML page. (#10)

  Show description for c500854
  balancap

  authored
  c500854

  Copy full SHA for c500854

Commits on Apr 26, 2023

• Update README.md following new IPU PjRt backend. (#24)

  Show description for bb9442f
  balancap

  authored
  bb9442f

  Copy full SHA for bb9442f

• Deactivate the use of tuple arguments on IPU backend. Not yet supported. (#23)

  balancap

  authored
  d890c1c

  Copy full SHA for d890c1c

• JAX IPU `pmap` and `pjit` quickstart notebooks and examples. (#22)

  Show description for 9be390a
  balancap

  authored
  9be390a

  Copy full SHA for 9be390a

Commits on Apr 18, 2023

• Switch JAX experimental on IPU to new PjRt client. (#21)

  Show description for d5f6640
  balancap

  authored
  d5f6640

  Copy full SHA for d5f6640

Commits on Apr 17, 2023

• Add IPU suffix to JAX Python wheel package. (#20)

  Show description for 8aad5ab
  balancap

  authored
  8aad5ab

  Copy full SHA for 8aad5ab

• Using XLA computation in MeshExecutable compilation. (#19)

  balancap

  committed
  31342ff

  Copy full SHA for 31342ff

Commits on Apr 5, 2023

• GCN tutorial notebook in IPU examples (#8)

  Show description for ec22f95
  danjust

  authored
  ec22f95

  Copy full SHA for ec22f95

Commits on Mar 24, 2023

• Add JAX IPU client flag, to choose backend implementation. (#17)

  Show description for e8fb8ab
  balancap

  committed
  e8fb8ab

  Copy full SHA for e8fb8ab

Commits on Feb 28, 2023

• Update README.md (#7)

  Show description for cc518ca
  awf

  authored
  cc518ca

  Copy full SHA for cc518ca

Commits on Feb 24, 2023

• Add JAX on IPU MNIST neural net training notebook. (#6)

  Show description for 6ba0e64
  balancap

  authored
  6ba0e64

  Copy full SHA for 6ba0e64

• Add JAX on IPU stateful linear regression example. (#5)

  Show description for 3ad9822
  balancap

  authored
  3ad9822

  Copy full SHA for 3ad9822

Commits on Feb 23, 2023

• Add execute permission to Paperspace setup.sh (#4)

  balancap

  authored
  fa85400

  Copy full SHA for fa85400

• Modify IPU JAX notebooks to be compatible with Paperspace Gradient. (#3)

  Show description for 7aaf28b
  balancap

  authored
  7aaf28b

  Copy full SHA for 7aaf28b

• Github CI compatible with public repository. (#2)

  Show description for 0a16dae
  balancap

  authored
  0a16dae

  Copy full SHA for 0a16dae

Commits on Feb 14, 2023

• Additional corrections to IPU documentation. (#11)

  balancap

  authored
  0e89ab2

  Copy full SHA for 0e89ab2

Commits on Feb 12, 2023

• Re-activate github action CI JAX IPU tests. (#10)

  balancap

  authored
  e0603fc

  Copy full SHA for e0603fc

Commits on Feb 7, 2023

• Add JAX license paragraph in README.md (#8)

  balancap

  authored
  246afd4

  Copy full SHA for 246afd4

Commits on Feb 6, 2023

• Fix LAX bitwise shift operators on IPU. (#7)

  Show description for b0ad504
  balancap

  authored
  b0ad504

  Copy full SHA for b0ad504

Commits on Feb 1, 2023

• Default eigh and LU algorithms lowering on IPU backend. (#6)

  Show description for e0845f6
  balancap

  authored
  e0845f6

  Copy full SHA for e0845f6

Commits on Jan 31, 2023

• Improve IPU JAX documentation and flags. (#5)

  Show description for 4b28c45
  

  balancap

  and

  awf

  authored
  4b28c45

  Copy full SHA for 4b28c45

Commits on Jan 28, 2023

• Fix JAX LAX sorting on IPU device. (#4)

  Show description for 26a23a1
  balancap

  authored
  26a23a1

  Copy full SHA for 26a23a1

Commits on Jan 27, 2023

• Add IPU MNIST training examples. (#3)

  balancap

  authored
  28fbce6

  Copy full SHA for 28fbce6

• Additional IPU documentation related to building and performance. (#2)

  balancap

  authored
  4d4cb7a

  Copy full SHA for 4d4cb7a

Commits on Jan 25, 2023

• IPU: Short readme for experimental JAX on IPU Mk2. (#1)

  Show description for ad7da96
  balancap

  committed
  ad7da96

  Copy full SHA for ad7da96

• IPU: Setup basic JAX CI action running unit tests. (#9)

  balancap

  committed
  05c6181

  Copy full SHA for 05c6181

• IPU: Using custom argminmax translation rule on IPU backend.

  

  li-zihang

  authored and

  balancap

  committed
  ea32cb6

  Copy full SHA for ea32cb6

Pagination

Previous

Next