Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 14, 2026

• Replace unicode escaped characters in ipynb files

  Show description for 9314161
  

  AleksMat

  authored and

  pax authors

  committed
  9314161

  Copy full SHA for 9314161

Commits on Dec 19, 2025

• [pmap] Fix unreplicate logic for new pmap.

  Show description for d340ef2
  

  danielsuo

  authored and

  pax authors

  committed
  d340ef2

  Copy full SHA for d340ef2

Commits on Dec 12, 2025

• [pmap] Create a local copy of `PmapSharding.default`.

  Show description for 1052d95
  

  danielsuo

  authored and

  pax authors

  committed
  1052d95

  Copy full SHA for 1052d95

Commits on Dec 8, 2025

• Add segment_pos to VisionTransformer.

  Show description for fc4318c
  

  Wojciech Sirko

  authored and

  pax authors

  committed
  fc4318c

  Copy full SHA for fc4318c

Commits on Nov 17, 2025

• Add `from __future__ import annotations` import to praxis linear layers.

  Show description for 4bcc337
  

  hawkinsp

  authored and

  pax authors

  committed
  4bcc337

  Copy full SHA for 4bcc337

Commits on Oct 24, 2025

• Remove unused `jax.lib.xla_client` import.

  Show description for 682f6a8
  

  Jake VanderPlas

  authored and

  pax authors

  committed
  682f6a8

  Copy full SHA for 682f6a8

Commits on Sep 25, 2025

• Make Pax quantizer configurable with custom quantization implementations

  Show description for fd2de54
  

  ablavatski

  authored and

  pax authors

  committed
  fd2de54

  Copy full SHA for fd2de54

Commits on Sep 2, 2025

• [praxis] Remove `jax_pmap_no_rank_reduction=False`.

  Show description for bd4e2bd
  

  danielsuo

  authored and

  pax authors

  committed
  bd4e2bd

  Copy full SHA for bd4e2bd

Commits on Jul 30, 2025

• Add logits_to_logp() method for SigmoidCrossEntropy.

  Show description for 1e1cfc9
  

  Krishna Sapkota

  authored and

  pax authors

  committed
  1e1cfc9

  Copy full SHA for 1e1cfc9

Commits on Jul 16, 2025

• Silence pytype errors related to improved `jax.nn` type annotations.

  Show description for 72dd7b6
  

  Jake VanderPlas

  authored and

  pax authors

  committed
  72dd7b6

  Copy full SHA for 72dd7b6

Commits on Jun 5, 2025

• Clean up unused value in Pax test_util.

  Show description for 41191d2
  

  The praxis Authors

  committed
  41191d2

  Copy full SHA for 41191d2

Commits on May 22, 2025

• Bump requirements.

  Show description for f5f1602
  

  laurentes

  authored and

  pax authors

  committed
  f5f1602

  Copy full SHA for f5f1602

Commits on Apr 28, 2025

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

  Show description for 68a4dbe
  

  bixia1

  authored and

  pax authors

  committed
  68a4dbe

  Copy full SHA for 68a4dbe

Commits on Apr 26, 2025

• Update requirements.txt

  Show description for 4f01af4
  

  laurentes

  authored and

  pax authors

  committed
  4f01af4

  Copy full SHA for 4f01af4

Commits on Apr 12, 2025

• Update pip requirements.

  Show description for 885be93
  

  laurentes

  authored and

  pax authors

  committed
  885be93

  Copy full SHA for 885be93

Commits on Apr 10, 2025

• Configure tests to explicitly use jax_threefry_partitionable=False.

  Show description for 2f8f1a6
  

  jburnim

  authored and

  pax authors

  committed
  2f8f1a6

  Copy full SHA for 2f8f1a6

Commits on Apr 9, 2025

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

  Show description for ac20ec0
  

  hyeontaek

  authored and

  pax authors

  committed
  ac20ec0

  Copy full SHA for ac20ec0

Commits on Apr 8, 2025

• Makes CLS init configurable.

  Show description for 8585099
  

  The praxis Authors

  committed
  8585099

  Copy full SHA for 8585099

Commits on Mar 26, 2025

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

  Show description for b237889
  

  Wojciech Sirko

  authored and

  pax authors

  committed
  b237889

  Copy full SHA for b237889

Commits on Feb 21, 2025

• [praxis] avoid use of jnp.size on tuples

  Show description for 45d6682
  

  Jake VanderPlas

  authored and

  pax authors

  committed
  45d6682

  Copy full SHA for 45d6682

Commits on Feb 5, 2025

• Use assertAllClose to compare float32 arrays

  Show description for ccd25a1
  

  ezhulenev

  authored and

  pax authors

  committed
  ccd25a1

  Copy full SHA for ccd25a1

Commits on Jan 29, 2025

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

  Show description for ccd78af
  

  oprypin

  authored and

  pax authors

  committed
  ccd78af

  Copy full SHA for ccd78af

Commits on Jan 24, 2025

• Sketch implementation of the sedd paper.

  Show description for d0e65fa
  

  The praxis Authors

  committed
  d0e65fa

  Copy full SHA for d0e65fa

Commits on Dec 20, 2024

• Update requirements

  Show description for 0350fa0
  

  laurentes

  authored and

  pax authors

  committed
  0350fa0

  Copy full SHA for 0350fa0

Commits on Dec 18, 2024

• Add visibility for Cloud Retail Search team.

  Show description for b29fd54
  

  The praxis Authors

  committed
  b29fd54

  Copy full SHA for b29fd54

Commits on Dec 16, 2024

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

  Show description for 820c4e2
  

  laurentes

  authored and

  pax authors

  committed
  820c4e2

  Copy full SHA for 820c4e2

Commits on Dec 14, 2024

• ...enable the fully_zero_masks and fix the decoding issue for local window size enabled...

  Show description for 91c0996
  

  The praxis Authors

  committed
  91c0996

  Copy full SHA for 91c0996

• ...In some multihead models, the inputs are shaped [batch, seq, dims]. In these cases, the code should treat the last two axes as the batch and sequence axes....

  Show description for edc9a6f
  

  The praxis Authors

  committed
  edc9a6f

  Copy full SHA for edc9a6f

Commits on Dec 6, 2024

• Bump requirements.

  Show description for ecff295
  

  laurentes

  authored and

  pax authors

  committed
  ecff295

  Copy full SHA for ecff295

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.

  Show description for 899b56e
  

  Jake VanderPlas

  authored and

  pax authors

  committed
  899b56e

  Copy full SHA for 899b56e

Commits on Nov 5, 2024

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

  Show description for 9693c08
  

  yueqiw

  authored and

  pax authors

  committed
  9693c08

  Copy full SHA for 9693c08

Commits on Oct 25, 2024

• Bump requirements

  Show description for ff73f49
  

  laurentes

  authored and

  pax authors

  committed
  ff73f49

  Copy full SHA for ff73f49

Commits on Oct 24, 2024

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

  Show description for 81aa951
  

  dougalm

  authored and

  pax authors

  committed
  81aa951

  Copy full SHA for 81aa951

Commits on Oct 17, 2024

• add scalar quantizer

  Show description for 572b571
  

  bignamehyp

  authored and

  pax authors

  committed
  572b571

  Copy full SHA for 572b571

Commits on Oct 6, 2024

• Merge pull request #69 from wenscarl:fp8_direct_quant

  Show description for 619fc34
  

  pax authors

  committed
  619fc34

  Copy full SHA for 619fc34

Pagination

Previous

Next