Search results

<code>sharding</code> parameter bug in JAX 0.4.31

bugSomething isn't workingSomething isn't working

Status: Open.

#132 In graphcore-research/jax-scalify;

· balancap opened on Aug 12, 2024

Reference documentation website

documentationImprovements or additions to documentationImprovements or additions to documentation

Status: Open.

#114 In graphcore-research/jax-scalify;

· balancap opened on Jun 26, 2024

Fix usability of <code>Scalify</code> in JAX eager mode

bugSomething isn't workingSomething isn't working

Status: Open.

#89 In graphcore-research/jax-scalify;

· balancap opened on Jan 17, 2024

Formalize minimal set of ops on a scale datatype

enhancementNew feature or requestNew feature or request

opsOps coverageOps coverage

Status: Open.

#64 In graphcore-research/jax-scalify;

· balancap opened on Jan 2, 2024

Stochastic rounding on <code>scalify</code> scale exponent operations

experimentsExperimentsExperiments

opsOps coverageOps coverage

Status: Open.

#63 In graphcore-research/jax-scalify;

· balancap opened on Jan 2, 2024

Support multi-devices JAX parallel operators <code>psum</code>, <code>pmean</code>, <code>pmin</code>, <code>pmax</code>

opsOps coverageOps coverage

Status: Open.

#24 In graphcore-research/jax-scalify;

· balancap opened on Nov 17, 2023

Generalize <code>Scalify</code> to support MX microscaling formats

enhancementNew feature or requestNew feature or request

Status: Open.

#21 In graphcore-research/jax-scalify;

· balancap opened on Nov 17, 2023