Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jun 4, 2026

• Update oryx to new JAX internal API

  Show description for 7e2470e
  

  dougalm

  authored and

  The oryx Authors

  committed
  7e2470e

  Copy full SHA for 7e2470e

Commits on May 29, 2026

• Delete not_mapped and replace it with None

  Show description for 28fe376
  

  yashk2810

  authored and

  The oryx Authors

  committed
  28fe376

  Copy full SHA for 28fe376

Commits on Apr 20, 2026

• Add a new stage_p primitive and Trace.stage_value operation.

  Show description for 5fa6d93
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  5fa6d93

  Copy full SHA for 5fa6d93

Commits on Apr 6, 2026

• Use the tracing registry when packing pytreedefs with FlatTree.

  Show description for 3985135
  

  rdyro

  authored and

  The oryx Authors

  committed
  3985135

  Copy full SHA for 3985135

Commits on Apr 1, 2026

• Update harvest_test after upstream changes including refactor of pmap to use shard_map.

  Show description for 8bd619e
  

  The oryx Authors

  committed
  8bd619e

  Copy full SHA for 8bd619e

Commits on Mar 31, 2026

• update oryx to match new scan_p bind signature

  Show description for 14f3fe5
  

  mattjj

  authored and

  The oryx Authors

  committed
  14f3fe5

  Copy full SHA for 14f3fe5

Commits on Mar 24, 2026

• Removed a few more unused internal APIs

  Show description for 26da298
  

  superbobry

  authored and

  The oryx Authors

  committed
  26da298

  Copy full SHA for 26da298

Commits on Mar 19, 2026

• Remove xla_pmap_p since it's deleted from JAX

  Show description for d491fba
  

  yashk2810

  authored and

  The oryx Authors

  committed
  d491fba

  Copy full SHA for d491fba

Commits on Mar 12, 2026

• Change Primitive.bind_with_trace to accept a list of avals.

  Show description for b39459e
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  b39459e

  Copy full SHA for b39459e

Commits on Mar 11, 2026

• [JAX] Require all positional arguments passed to bind to be JAX values.

  Show description for ee202ab
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  ee202ab

  Copy full SHA for ee202ab

Commits on Feb 27, 2026

• Internal change

  Show description for 2469b7f
  

  danielsuo

  authored and

  The oryx Authors

  committed
  2469b7f

  Copy full SHA for 2469b7f

Commits on Jan 9, 2026

• Fix FlatTree usage in oryx

  Show description for 4216cde
  

  yashk2810

  authored and

  The oryx Authors

  committed
  4216cde

  Copy full SHA for 4216cde

Commits on Dec 12, 2025

• Update oryx to new FlatTree-based trace_to_jaxpr api

  Show description for aa33090
  

  dougalm

  authored and

  The oryx Authors

  committed
  aa33090

  Copy full SHA for aa33090

Commits on Dec 9, 2025

• Replace deprecated `jax.core.get_aval` with `jax.typeof`

  Show description for 06ce7e9
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  06ce7e9

  Copy full SHA for 06ce7e9

Commits on Dec 4, 2025

• PR #33710: [no-thunks] Deprecate old form of `cond` and remove some tracing layers.

  Show description for e6823b2
  

  dougalm

  authored and

  The oryx Authors

  committed
  e6823b2

  Copy full SHA for e6823b2

Commits on Sep 25, 2025

• Disable `pmap` tests in `InverseTest` under `pmap_shmap_merge=True`.

  Show description for ef8a65f
  

  danielsuo

  authored and

  The oryx Authors

  committed
  ef8a65f

  Copy full SHA for ef8a65f

Commits on Sep 18, 2025

• Make the commented-out x64_context_test tests pass.

  Show description for 2662061
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  2662061

  Copy full SHA for 2662061

Commits on Sep 6, 2025

• [debug_info] Fixing more places where debug information may be wrong

  Show description for a54f729
  

  gnecula

  authored and

  The oryx Authors

  committed
  a54f729

  Copy full SHA for a54f729

Commits on Aug 20, 2025

• [oryx] avoid use of deprecated partial_eval.Jaxpr

  Show description for 4ac1910
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  4ac1910

  Copy full SHA for 4ac1910

Commits on Aug 19, 2025

• [oryx] Switch to internal partial_eval APIs

  Show description for bea7486
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  bea7486

  Copy full SHA for bea7486

Commits on Aug 18, 2025

• [oryx] Switch to internal ad APIs

  Show description for 4103b6f
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  4103b6f

  Copy full SHA for 4103b6f

• Switch to internal batching APIs

  Show description for 941b880
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  941b880

  Copy full SHA for 941b880

Commits on Aug 13, 2025

• Replace usage of JAX's private `tree_util._replace_nones`.

  Show description for 6b4ff9a
  

  jburnim

  authored and

  The oryx Authors

  committed
  6b4ff9a

  Copy full SHA for 6b4ff9a

Commits on Jul 22, 2025

• Use `jnp.zeros_like` instead of the internal `lax.zeros_like_array`

  Show description for cb95be2
  

  superbobry

  authored and

  The oryx Authors

  committed
  cb95be2

  Copy full SHA for cb95be2

Commits on Jul 10, 2025

• remove `pjit_p`, leaving only `jax.extend.core.primitives.jit_p`

  Show description for 5fc482a
  

  froystig

  authored and

  The oryx Authors

  committed
  5fc482a

  Copy full SHA for 5fc482a

Commits on Jun 16, 2025

• [JAX] Make parameters to jaxpr equations hashable.

  Show description for 2619298
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  2619298

  Copy full SHA for 2619298

Commits on May 30, 2025

• [JAX] replace uses of internal jax/_src/pjit.py and //third_party/py/jax:pjit

  Show description for a665264
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  a665264

  Copy full SHA for a665264

Commits on May 13, 2025

• [Oryx] Fixes reap/plant rules for cond in harvest

  Show description for f3b7c8e
  

  gnecula

  authored and

  The oryx Authors

  committed
  f3b7c8e

  Copy full SHA for f3b7c8e

Commits on May 8, 2025

• [oryx] Replace uses of deprecated jax.util with jax._src.util.

  Show description for 5be8032
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  5be8032

  Copy full SHA for 5be8032

Commits on Apr 25, 2025

• fix scan `linear=` length issue, exposed by #28210

  Show description for 8fe26a9
  

  mattjj

  authored and

  The oryx Authors

  committed
  8fe26a9

  Copy full SHA for 8fe26a9

Commits on Apr 17, 2025

• Remove shard_map rewrite machinery since we have `vma` in types by default

  Show description for cb8cf48
  

  yashk2810

  authored and

  The oryx Authors

  committed
  cb8cf48

  Copy full SHA for cb8cf48

Commits on Mar 28, 2025

• Add accuracy field to unary ops

  Show description for 4ca17fe
  

  hanrach9

  authored and

  The oryx Authors

  committed
  4ca17fe

  Copy full SHA for 4ca17fe

Commits on Mar 13, 2025

• [JAX] Change jax.core.Trace subclasses to call super().__init__().

  Show description for d2c6862
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  d2c6862

  Copy full SHA for d2c6862

Commits on Mar 5, 2025

• Update references to JAX's GitHub repo

  Show description for ba65ee7
  

  jakeharmon8

  authored and

  The oryx Authors

  committed
  ba65ee7

  Copy full SHA for ba65ee7

Commits on Feb 13, 2025

• Update oryx for JAX debug_info

  Show description for 0156890
  

  gnecula

  authored and

  The oryx Authors

  committed
  0156890

  Copy full SHA for 0156890