8000 Commits · google/praxis · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on May 22, 2025

Bump requirements.

laurentes
authored and
pax authors
committed

Commits on Apr 28, 2025

[JAX] Replace GSPMDSharding with NamedSharding since GSPMDSHarding is deprecated in JAX and will be removed in future JAX releases

bixia1
authored and
pax authors
committed

Commits on Apr 26, 2025

Update requirements.txt

laurentes
authored and
pax authors
committed

Commits on Apr 12, 2025

Update pip requirements.

laurentes
authored and
pax authors
committed

Commits on Apr 10, 2025

Configure tests to explicitly use jax_threefry_partitionable=False.

jburnim
authored and
pax authors
committed

Commits on Apr 9, 2025

[Praxis] Use `fp8_ops.fp8_scaled_dot_general` from the latest flax.linen

hyeontaek
authored and
pax authors
committed

Commits on Apr 8, 2025

Makes CLS init configurable.
The praxis Authors
committed

Commits on Mar 26, 2025

Update feed_forward_tpl type to reflect the fact that None is supported.

Wojciech Sirko
authored and
pax authors
committed

Commits on Feb 21, 2025

[praxis] avoid use of jnp.size on tuples

Jake VanderPlas
authored and
pax authors
committed

Commits on Feb 5, 2025

Use assertAllClose to compare float32 arrays

ezhulenev
authored and
pax authors
committed

Commits on Jan 29, 2025

Remove `srcs_version` and `python_version` attributes, as they already default to `"PY3"`

oprypin
authored and
pax authors
committed

Commits on Jan 24, 2025

Sketch implementation of the sedd paper.
The praxis Authors
committed

Commits on Dec 20, 2024

Update requirements

laurentes
authored and
pax authors
committed

Commits on Dec 18, 2024

Add visibility for Cloud Retail Search team.
The praxis Authors
committed

Commits on Dec 16, 2024

Bump Praxis OSS pre-/post-submits timeout by 50%

laurentes
authored and
pax authors
committed

Commits on Dec 14, 2024

Commits on Dec 6, 2024

Bump requirements.

laurentes
authored and
pax authors
committed

Commits on Nov 11, 2024

In jnp.reshape, use `shape` rather than deprecated `newshape`. The `newshape` parameter was deprecated in JAX v0.4.28, and will soon be removed.

Jake VanderPlas
authored and
pax authors
committed

Commits on Nov 5, 2024

Add an option to shift output ids by an offset in beam search

yueqiw
authored and
pax authors
committed

Commits on Oct 25, 2024

Bump requirements

laurentes
authored and
pax authors
committed

Commits on Oct 24, 2024

Avoid depending on JAX internals, which are about to change.

dougalm
authored and
pax authors
committed

Commits on Oct 17, 2024

add scalar quantizer

bignamehyp
authored and
pax authors
committed

Commits on Oct 6, 2024

Merge pull request #69 from wenscarl:fp8_direct_quant
pax authors
committed

Commits on Oct 4, 2024

Implement SequenceLangaugeDPO model.
The praxis Authors
committed
Bump requirements incl. jaxlib

laurentes
authored and
pax authors
committed
Fix type mismatch error for 3DCNN.

jianlijianli
authored and
pax authors
committed

Commits on Oct 2, 2024

Support fp8 direct quantization for Einsum and Gated Einsum.
wenscarl
committed
Fix function name following https://github.com/google/flax/pull/4229

laurentes
authored and
pax authors
committed

Commits on Sep 23, 2024

Test VQVAE.

jianlijianli
authored and
pax authors
committed

Commits on Sep 22, 2024

Merge pull request #85 from kaixih:support_gqa
pax authors
committed

Commits on Sep 19, 2024

Add unit test for binary vq.

jianlijianli
authored and
pax authors
committed

Commits on Sep 17, 2024

Bump requirements.

laurentes
authored and
pax authors
committed

Commits on Sep 12, 2024

Update requirements.

laurentes
authored and
pax authors
committed

Commits on Sep 5, 2024

internal
The praxis Authors
committed

0