Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 28, 2023

• Move configs to a separate file.

  Show description for a54f8e0
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  a54f8e0

  Copy full SHA for a54f8e0

Commits on Apr 27, 2023

• Move use_fake_quant to the config.

  Show description for abc2acd
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  abc2acd

  Copy full SHA for abc2acd

• Make dimension_numbers static.

  Show description for 22e6d92
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  22e6d92

  Copy full SHA for 22e6d92

• Convert nondiff_arguments to "None-gradient" arguments.

  Show description for ca439da
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  ca439da

  Copy full SHA for ca439da

Commits on Apr 26, 2023

• Make context a keyword only argument. Bugfix: preserve lax.dot_general positional argument order.

  Show description for 455f1ed
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  455f1ed

  Copy full SHA for 455f1ed

• Silence some pytype errors.

  Show description for 714b410
  

  rchen152

  authored and

  copybara-github

  committed
  714b410

  Copy full SHA for 714b410

Commits on Apr 25, 2023

• Adds sparsity tests for pruning functionality to praxis layers sparsity.

  Show description for 0cf7c8c
  

  shivaniag

  authored and

  copybara-github

  committed
  0cf7c8c

  Copy full SHA for 0cf7c8c

Commits on Apr 24, 2023

• Move new AQT code to vs.

  Show description for cdbdc49
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  cdbdc49

  Copy full SHA for cdbdc49

• Several incremental changes.

  Show description for 0571781
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  0571781

  Copy full SHA for 0571781

Commits on Apr 21, 2023

• Factor the key out into a context dataclass.

  Show description for 7c227d8
  

  ychzhang

  authored and

  copybara-github

  committed
  7c227d8

  Copy full SHA for 7c227d8

Commits on Apr 20, 2023

• Make config constructors into class methods "make"; Move check_eq out of test.

  Show description for 340ab73
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  340ab73

  Copy full SHA for 340ab73

Commits on Apr 19, 2023

• Ensure argument is positive before it compares with clip_range().

  Show description for ff62542
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  ff62542

  Copy full SHA for ff62542

Commits on Apr 13, 2023

• The default API now attaches custom gradients:

  Show description for 6c2c538
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  6c2c538

  Copy full SHA for 6c2c538

• Remove custom gradients added to matmuls when generating serving graphs since those cause the float weights to get pulled in due to some silly ops added at the output of the function with the custo…

  Show description for e6e5b28
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  e6e5b28

  Copy full SHA for e6e5b28

Commits on Apr 12, 2023

• Remove unused float weights from the serving graph while using AQT.

  Show description for c9c67a9
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  c9c67a9

  Copy full SHA for c9c67a9

• Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

  Show description for 4bf888d
  

  chiamp

  authored and

  copybara-github

  committed
  4bf888d

  Copy full SHA for 4bf888d

Commits on Apr 7, 2023

• Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

  Show description for 765d6ee
  

  chiamp

  authored and

  copybara-github

  committed
  765d6ee

  Copy full SHA for 765d6ee

• Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

  Show description for 764116b
  

  chiamp

  authored and

  copybara-github

  committed
  764116b

  Copy full SHA for 764116b

• Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

  Show description for de9c1bc
  

  chiamp

  authored and

  copybara-github

  committed
  de9c1bc

  Copy full SHA for de9c1bc

Commits on Apr 4, 2023

• Stochastic rounding.

  Show description for 6a9a827
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  6a9a827

  Copy full SHA for 6a9a827

Commits on Mar 30, 2023

• Test for hardware int8

  Show description for 07ad77e
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  07ad77e

  Copy full SHA for 07ad77e

• use_hardware_int8

  Show description for 843b9a3
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  843b9a3

  Copy full SHA for 843b9a3

Commits on Mar 29, 2023

• Add quantized training (backprop quantization).

  Show description for b458abc
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  b458abc

  Copy full SHA for b458abc

• floor does not have a custom gradient anymore.

  Show description for bac71fe
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  bac71fe

  Copy full SHA for bac71fe

Commits on Mar 16, 2023

• [JAX] Update some type annotations to fix some pytype errors.

  Show description for f76f92a
  

  hawkinsp

  authored and

  copybara-github

  committed
  f76f92a

  Copy full SHA for f76f92a

Commits on Mar 14, 2023

• [JAX] Replace uses of deprecated DeviceArray class with jax.Array.

  Show description for 9fe75fe
  

  hawkinsp

  authored and

  copybara-github

  committed
  9fe75fe

  Copy full SHA for 9fe75fe

• Replaces references to jax.numpy.DeviceArray with jax.Array.

  Show description for 43b14a6
  

  hawkinsp

  authored and

  copybara-github

  committed
  43b14a6

  Copy full SHA for 43b14a6

Commits on Mar 9, 2023

• Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

  Show description for bcdafcc
  

  chiamp

  authored and

  copybara-github

  committed
  bcdafcc

  Copy full SHA for bcdafcc

Commits on Feb 24, 2023

• Internal

  Show description for 4d5c9cd
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  4d5c9cd

  Copy full SHA for 4d5c9cd

Commits on Feb 23, 2023

• Internal

  Show description for 6b471ed
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  6b471ed

  Copy full SHA for 6b471ed

Commits on Feb 22, 2023

• Bugfix dot_general, better tests.

  Show description for ac8ca01
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  ac8ca01

  Copy full SHA for ac8ca01

Commits on Feb 21, 2023

• Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

  Show description for ac03220
  

  hawkinsp

  authored and

  copybara-github

  committed
  ac03220

  Copy full SHA for ac03220

• Simplify dot_general.

  Show description for 100d8d8
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  100d8d8

  Copy full SHA for 100d8d8

• Quantization now is optional. Simplify fresh_scale logic.

  Show description for 1dfbe58
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  1dfbe58

  Copy full SHA for 1dfbe58

• Optimize rescaling in dot_general.

  Show description for f2dcb2b
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  f2dcb2b

  Copy full SHA for f2dcb2b

Pagination

Previous

Next