Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

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

Commits on Feb 6, 2025

• PR #26348: [better_errors] Continue adding debug info to Jaxprs (step 3)

  Show description for 9658f42
  

  gnecula

  authored and

  The oryx Authors

  committed
  9658f42

  Copy full SHA for 9658f42

Commits on Jan 27, 2025

• PR #26097: [better_errors] Refactor more uses of pe.tracing_debug_info (part 1)

  Show description for 4fe968c
  

  gnecula

  authored and

  The oryx Authors

  committed
  4fe968c

  Copy full SHA for 4fe968c

Commits on Dec 18, 2024

• Migrate from jax.core to jax.extend.core for several deprecated symbols

  Show description for d5d116c
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  d5d116c

  Copy full SHA for d5d116c

Commits on Dec 17, 2024

• PR #24970: Add a new API jax.lax.split.

  Show description for cba9837
  

  hawkinsp

  authored and

  The oryx Authors

  committed
  cba9837

  Copy full SHA for cba9837

Commits on Dec 13, 2024

• [JAX] Replace `jax.core.call_p` and `jax.core.closed_call_p` with `jax.extend.core.primitives.call_p` and `jax.extend.core.primitives.closed_call_p`.

  Show description for 4e93992
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  4e93992

  Copy full SHA for 4e93992

Commits on Dec 11, 2024

• Remove references to jax.core.raise_to_shaped

  Show description for 4ae0aaa
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  4ae0aaa

  Copy full SHA for 4ae0aaa

Commits on Dec 10, 2024

• Update minimum python version to match that of newly required JAX version.

  Show description for 32f02ef
  

  ColCarroll

  authored and

  The oryx Authors

  committed
  32f02ef

  Copy full SHA for 32f02ef

Commits on Dec 9, 2024

• Bumping jax dependency and oryx version to prepare release.

  Show description for 81fd0d7
  

  ColCarroll

  authored and

  The oryx Authors

  committed
  81fd0d7

  Copy full SHA for 81fd0d7

Commits on Nov 15, 2024

• Bump Oryx version to 0.2.8

  Show description for 33f7c4c
  

  ColCarroll

  authored and

  The oryx Authors

  committed
  33f7c4c

  Copy full SHA for 33f7c4c

Commits on Oct 29, 2024

• Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post…

  Show description for 4627e7d
  

  dougalm

  authored and

  The oryx Authors

  committed
  4627e7d

  Copy full SHA for 4627e7d

Commits on Oct 25, 2024

• Remove uses of jax._src.sharding_impls.is_unspecified internal API.

  Show description for 339bc92
  

  Jake VanderPlas

  authored and

  The oryx Authors

  committed
  339bc92

  Copy full SHA for 339bc92

Commits on Sep 18, 2024

• Clean up and fix primal type to tangent type mapping

  Show description for 5272a3d
  

  dougalm

  authored and

  The oryx Authors

  committed
  5272a3d

  Copy full SHA for 5272a3d

Commits on Aug 29, 2024

• Add basic support for shard_map to oryx

  Show description for 93fc9ef
  

  The oryx Authors

  committed
  93fc9ef

  Copy full SHA for 93fc9ef

Commits on Jun 27, 2024

• [Oryx] Make `linear` an optional parameter for cond_p.

  Show description for 21722fa
  

  dfm

  authored and

  The oryx Authors

  committed
  21722fa

  Copy full SHA for 21722fa

Commits on May 23, 2024

• Bump version to 0.2.7

  Show description for e29141b
  

  ColCarroll

  authored and

  The oryx Authors

  committed
  e29141b

  Copy full SHA for e29141b

Pagination

Previous

Next