graphcore-research/jax-scalify

Issues

• `sharding` parameter bug in JAX 0.4.31

  #132 opened 4 months ago by balancap
  0

• Fix import bug regression with JAX `<0.4.28`

  #121 opened 5 months ago by balancap
  0

• Add implementation of `jax_scalify.tree` sub-module.

  #115 opened 5 months ago by balancap
  0

• Reference documentation website

  #114 opened 5 months ago by balancap
  0

• Fix usability of `Scalify` in JAX eager mode

  #89 opened a year ago by balancap
  0

• Stochastic rounding on `scalify` scale exponent operations

  #63 opened a year ago by balancap
  0

• Generalize `Scalify` to support MX microscaling formats

  #21 opened a year ago by balancap
  5

• Training setup for 3 MLP layers

  #37 opened 6 months ago by balancap
  0

• Gaussian ops analysis notebook

  #17 opened 6 months ago by balancap
  0

• Floating point scale degrades FP16 MNIST training accuracy

  #60 opened 10 months ago by balancap
  3

• Bug in `jax.scipy.special.logsumexp` scale propagation

  #88 opened 10 months ago by balancap
  1

• Improve scale propagation in broadcasted scalar tensors.

  #94 opened 10 months ago by balancap
  1

• MNIST training example in FP8

  #62 opened a year ago by balancap
  2

• Use FP32 scale in MNIST training example

  #77 opened a year ago by balancap
  1

• MNIST training broken with `min/max` scale propagation

  #69 opened a year ago by balancap
  2

• Formalize minimal set of ops on a scale datatype

  #64 opened a year ago by balancap
  1

• Implement power of 2 scaling in `AutoScale`

  #61 opened a year ago by balancap
  0

• MNIST 2 layers MLP model training using AutoScale

  #18 opened a year ago by balancap
  1

• Support JAX (experimental) IPU version

  #38 opened a year ago by balancap
  0

• Implement basic boolean operators `eq`, `ne`, ...

  #27 opened a year ago by balancap
  0

• Support mixed normal/scaled graph in `AutoScale`

  #22 opened a year ago by balancap
  0

• Implement `reduce_sum/prod/max/min` scaled translation rules

  #23 opened a year ago by balancap
  0

• Support multi-devices JAX parallel operators `psum`, `pmean`, `pmin`, `pmax`

  #24 opened a year ago by balancap
  0