Commits

Branch selector

 main

User selector

superbobry

All time

Commit History

Commits on Dec 15, 2023

• Rolling back #18980, because it is not backwards compatible and breaks existing users.

  Show description for 4153112
  

  superbobry

  authored and

  jax authors

  committed
  4153112

  Copy full SHA for 4153112

Commits on Dec 13, 2023

• Upgrade remaining sources to Python 3.9

  Show description for f936613
  superbobry

  committed
  f936613

  Copy full SHA for f936613

Commits on Dec 12, 2023

• The pretty printer now de-duplicates identical jaxprs

  Show description for 840abfb
  superbobry

  committed
  840abfb

  Copy full SHA for 840abfb

Commits on Dec 11, 2023

• `Effects` is now an immutable set

  Show description for 352e10e
  

  superbobry

  authored and

  jax authors

  committed
  352e10e

  Copy full SHA for 352e10e

Commits on Dec 8, 2023

• Upgrade most .py sources to 3.9

  Show description for 36f6b52
  superbobry

  committed
  36f6b52

  Copy full SHA for 36f6b52

Commits on Dec 7, 2023

• Print pjit name= before other params

  Show description for ea158d3
  superbobry

  committed
  ea158d3

  Copy full SHA for ea158d3

Commits on Oct 29, 2023

• Deprecated `define_*` and `DEFINE_*` methods on `jax.config`

  Show description for fd3a8b2
  superbobry

  committed
  fd3a8b2

  Copy full SHA for fd3a8b2

Commits on Oct 27, 2023

• Deprecated accessing `config` via the `jax.config` submodule

  Show description for c90f1f0
  superbobry

  committed
  c90f1f0

  Copy full SHA for c90f1f0

Commits on Oct 23, 2023

• MAINT Do not use `str()` and `repr()` in f-string replacement fields

  Show description for f2ce5db
  superbobry

  committed
  f2ce5db

  Copy full SHA for f2ce5db

Commits on Oct 18, 2023

• MAINT Do not import the config object in JAX internals

  Show description for 1079304
  superbobry

  committed
  1079304

  Copy full SHA for 1079304

Commits on Oct 13, 2023

• MAINT Drop underscore from the name of externally-referenced state objects

  superbobry

  committed
  f9087ab

  Copy full SHA for f9087ab

Commits on Oct 12, 2023

• MAINT Migrate remaining internal/test modules to use state objects

  Show description for cbcaac2
  superbobry

  committed
  cbcaac2

  Copy full SHA for cbcaac2

Commits on Oct 11, 2023

• Migrate another subset of internal modules to use state objects

  Show description for 2f70ae7
  

  superbobry

  authored and

  jax authors

  committed
  2f70ae7

  Copy full SHA for 2f70ae7

Commits on Oct 10, 2023

• MAINT Use a generator expression with `all()` and `any()`

  Show description for 5d9c39f
  superbobry

  committed
  5d9c39f

  Copy full SHA for 5d9c39f

Commits on Oct 9, 2023

• Migrate a subset of internal modules to use state objects

  Show description for 65d3058
  

  superbobry

  authored and

  jax authors

  committed
  65d3058

  Copy full SHA for 65d3058

Commits on Oct 4, 2023

• `_StateContextManager` now preserves the type of the value it stores.

  Show description for 923498f
  

  superbobry

  authored and

  jax authors

  committed
  923498f

  Copy full SHA for 923498f

Commits on Oct 1, 2023

• MAINT Clean up leftover `Array = Any` aliases in jax/_src/**.py

  Show description for 5ab05e4
  superbobry

  committed
  5ab05e4

  Copy full SHA for 5ab05e4

Commits on Sep 28, 2023

• MAINT Reorder the overloads for `lax.sort`

  Show description for a8b8267
  superbobry

  committed
  a8b8267

  Copy full SHA for a8b8267

Commits on Sep 25, 2023

• ENH Use {} and () instead of dict() and tuple()

  superbobry

  committed
  eca10f5

  Copy full SHA for eca10f5

Commits on Sep 21, 2023

• MAINT Use a generator expression in tuple([... for ... in ...])

  Show description for df7f6a0
  superbobry

  committed
  df7f6a0

  Copy full SHA for df7f6a0

Commits on Sep 20, 2023

• DOC: remove redundant backticks in pjit tutorial stub

  superbobry

  committed
  1dbf863

  Copy full SHA for 1dbf863

Commits on Jul 4, 2023

• Removed the type annotation of `orig_out_sharding_handlers`

  Show description for 18fba6b
  

  superbobry

  authored and

  jax authors

  committed
  18fba6b

  Copy full SHA for 18fba6b

Commits on May 23, 2022

• MAINT Fixed new mypy errors

  Show description for c5d3ece
  superbobry

  committed
  c5d3ece

  Copy full SHA for c5d3ece

• ENH _wraps() in jax._src.numpy.util is now returns a generic function

  Show description for be14098
  superbobry

  committed
  be14098

  Copy full SHA for be14098

Commits on Apr 25, 2022

• compile_or_get_cached() is now more type-checker friendly

  Show description for 0483483
  superbobry

  authored
  0483483

  Copy full SHA for 0483483

Commits on Mar 31, 2022

• Suppressed a few more errors in jax/interpreters/mlir.py

  superbobry

  committed
  19c6657

  Copy full SHA for 19c6657

Commits on Mar 30, 2022

• Fixed a typo in the return type of _array_ir_types

  Show description for b315212
  superbobry

  authored
  b315212

  Copy full SHA for b315212

Commits on Feb 8, 2022

• Added an explicit Any return type to lax_numpy.ndarray methods

  Show description for 0fe377c
  superbobry

  committed
  0fe377c

  Copy full SHA for 0fe377c

Commits on Aug 16, 2021

• Added jaxlib to .pre-commit-config.yaml

  superbobry

  committed
  12f9a80

  Copy full SHA for 12f9a80

• Most of JAX now uses concrete types for things defined in jaxlib.xla_client

  Show description for af41a95
  superbobry

  committed
  af41a95

  Copy full SHA for af41a95

Commits on Aug 2, 2021

• Bumped TensorFlow revision to include xla_extension type stub fixes

  superbobry

  committed
  a204f54

  Copy full SHA for a204f54

• Type stubs for jaxlib.xla_extension no longer use -stubs suffix

  Show description for 2a994bd
  superbobry

  committed
  2a994bd

  Copy full SHA for 2a994bd

Commits on Apr 12, 2021

• Fixed xla_extension-stubs packaging

  Show description for 0ffd554
  superbobry

  committed
  0ffd554

  Copy full SHA for 0ffd554

Commits on Apr 6, 2021

• Re-exported tensorflow...xla_extension type stubs in jaxlib

  Show description for 225ffc3
  superbobry

  committed
  225ffc3

  Copy full SHA for 225ffc3

Commits on May 19, 2020

• Exported lax from jax/__init__.py (#3135)

  Show description for 73b76e9
  superbobry

  authored
  73b76e9

  Copy full SHA for 73b76e9