Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 19, 2023

• Updated jax.config import

  Show description for 73a4a75
  

  superbobry

  authored and

  tensorflower-gardener

  committed
  73a4a75

Commits on Sep 19, 2023

• Migrate TFP to support JAX typed PRNG keys

  Show description for 6efcda9
  

  vanderplas

  authored and

  tensorflower-gardener

  committed
  6efcda9

Commits on Aug 26, 2023

• FunMC: Make maybe_broadcast_structure more general.

  Show description for 0ff4e3a
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  0ff4e3a

Commits on Aug 15, 2023

• Remove internal dependency.

  Show description for 4d23c6a
  

  jburnim

  authored and

  tensorflower-gardener

  committed
  4d23c6a

Commits on Aug 4, 2023

• FunMC: AIS fixes.

  Show description for 3c563e3
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  3c563e3

Commits on Jul 27, 2023

• Internal Code Change

  Show description for 5ef0c6c
  

  rickeylev

  authored and

  tensorflower-gardener

  committed
  5ef0c6c

Commits on May 11, 2023

• Fix a few tests in FunMC/InferenceGym/discussion directory to work with jax_enable_custom_prng.

  Show description for 6a43309
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  6a43309

Commits on Apr 14, 2023

• FunMC: Update type annotations to Python 3.9

  Show description for d87ef8d
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  d87ef8d

Commits on Mar 2, 2023

• add default applicable_licenses

  Show description for 172e0e0
  

  Googler

  authored and

  tensorflower-gardener

  committed
  172e0e0

Commits on Feb 23, 2023

• FunMC: Add interruptible trace.

  Show description for 23f6c22
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  23f6c22

Commits on Feb 17, 2023

• Internal change.

  Show description for 5639d0e
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  5639d0e

Commits on Feb 14, 2023

• Internal change.

  Show description for 3097fc0
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  3097fc0

• add default applicable_licenses

  Show description for d37d0c4
  

  Googler

  authored and

  tensorflower-gardener

  committed
  d37d0c4

Commits on Feb 3, 2023

• Avoid capturing tracers in custom gradient implementations.

  Show description for c0908ab
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  c0908ab

Commits on Dec 4, 2022

• FunMC: Add AIS/SMC.

  Show description for e816859
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  e816859

Commits on Oct 10, 2022

• Use the jax.Array public type rather than the priviate jax.xla.DeviceArray type.

  Show description for 39aee75
  

  yashk2810

  authored and

  tensorflower-gardener

  committed
  39aee75

Commits on Sep 30, 2022

• Malt: Add grid searches, fix the eigenvalue computation, fix damping.

  Show description for 2bf8e4b
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  2bf8e4b

Commits on Sep 29, 2022

• Remove tf.test.TestCase dependency for tests.

  Show description for 93cd54a
  

  srvasude

  authored and

  tensorflower-gardener

  committed
  93cd54a

Commits on Sep 28, 2022

• Replace jax.xla.DeviceArray private type with the new public type jax.Array.

  Show description for 51bb20b
  

  yashk2810

  authored and

  tensorflower-gardener

  committed
  51bb20b

Commits on Aug 15, 2022

• FunMC: Use Python 3.7-compatible constructs since external Colab only supports 3.7.

  Show description for b13dffa
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  b13dffa

Commits on Jul 27, 2022

• [Oryx] Remove Oryx from TFP (since it now lives in github.com/jax-ml/oryx).

  Show description for 796ecd6
  

  sharadmv

  authored and

  tensorflower-gardener

  committed
  796ecd6

Commits on Jul 21, 2022

• Change type annonation of 'tag' in 'sow' to Hashable to allow for Enums as tags. Add tests for hashable reap & plant.

  Show description for 4153e36
  

  armanroshannai

  authored and

  tensorflower-gardener

  committed
  4153e36

• Unify TF AutoCompositeTensor and JAX Pytree logic in TFP. Fix JointDistributions as Pytrees.

  Show description for c2e07c7
  

  emilyfertig

  authored and

  tensorflower-gardener

  committed
  c2e07c7

• FunMC: Implement MALT

  Show description for 6663d2c
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  6663d2c

Commits on Jul 20, 2022

• Fix data dependency issue in harvest.sow when reaping and planting constants. Add test cases.

  Show description for ea99779
  

  armanroshannai

  authored and

  tensorflower-gardener

  committed
  ea99779

Commits on Jul 15, 2022

• Use jax.tree_util.tree_map in place of deprecated tree_multimap.

  Show description for cc3a6bd
  

  vanderplas

  authored and

  tensorflower-gardener

  committed
  cc3a6bd

• FunMC: Allow specifying Adam kwargs for step size adaptation.

  Show description for 42cb674
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  42cb674

Commits on Jul 14, 2022

• Fix data dependency issue in harvest.sow when reaping and planting constants. Add test cases.

  Show description for 275495d
  

  armanroshannai

  authored and

  tensorflower-gardener

  committed
  275495d

Commits on Jun 17, 2022

• [JAX] Remove references to omnistaging.

  Show description for cb2a94c
  

  hawkinsp

  authored and

  tensorflower-gardener

  committed
  cb2a94c

Commits on May 30, 2022

• Improve Readability

  Show description for 1166fe9
  ryanrussell

  committed
  1166fe9

Commits on May 24, 2022

• FunMC: Use tree_util.tree_map instead of tree_util.tree_multimap to silence warning.

  Show description for 3a6f248
  

  SiegeLordEx

  authored and

  tensorflower-gardener

  committed
  3a6f248

Commits on May 4, 2022

• [Oryx] Move extract_call_jaxpr to trace_util and remove usage of unit.

  Show description for feb28a2
  

  sharadmv

  authored and

  tensorflower-gardener

  committed
  feb28a2

Commits on May 3, 2022

• [Oryx] Delete the unzip transformation

  Show description for fae5c70
  

  sharadmv

  authored and

  tensorflower-gardener

  committed
  fae5c70

Commits on Apr 29, 2022

• [Oryx] Fix unzip to use updated JaxprEqnRecipe fields

  Show description for 485cd10
  

  sharadmv

  authored and

  tensorflower-gardener

  committed
  485cd10

Commits on Apr 21, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for c12f626
  

  yilei

  authored and

  tensorflower-gardener

  committed
  c12f626

Pagination

PreviousNext