Commits

Branch selector

main

User selector

All users

All time

Commit history

Commits on Oct 27, 2021

• Updates examples/imagenet README.

  Show description for 8dfcd63
  

  andsteing

  authored and

  Flax Authors

  committed
  8dfcd63View commit details

  Copy full SHA for 8dfcd63

  Browse repository at this point

Commits on Oct 25, 2021

• Merge pull request #1629 from jheek:lifted-jacobian-products

  Show description for 03ea1a7
  

  Flax Authors

  committed
  03ea1a7View commit details

  Copy full SHA for 03ea1a7

  Browse repository at this point

• Merge pull request #1640 from jheek/add-nowrap-rtd

  Show description for 1e2a6c3
  jheek

  authored
  1e2a6c3View commit details

  Copy full SHA for 1e2a6c3

  Browse repository at this point

• Fix docs

  jheek

  committed
  0a1c890View commit details

  Copy full SHA for 0a1c890

  Browse repository at this point

• Fix pytype issue

  jheek

  committed
  31d04f8View commit details

  Copy full SHA for 31d04f8

  Browse repository at this point

• Add nowrap to RTD

  jheek

  committed
  e62c7ccView commit details

  Copy full SHA for e62c7cc

  Browse repository at this point

Commits on Oct 20, 2021

• Implement vjp and jvp transforms

  jheek

  committed
  6906619View commit details

  Copy full SHA for 6906619

  Browse repository at this point

Commits on Oct 18, 2021

• Fixes intendation for some code snippets in `errors.py`.

  Show description for 38e6bda
  

  andsteing

  authored and

  Flax Authors

  committed
  38e6bdaView commit details

  Copy full SHA for 38e6bda

  Browse repository at this point

• Switch ogbg_molpcba header image to PNG.

  Show description for 7eee5b0
  

  ameya98

  authored and

  Flax Authors

  committed
  7eee5b0View commit details

  Copy full SHA for 7eee5b0

  Browse repository at this point

• Merge pull request #1623 from jheek:fix-partial-module-method

  Show description for 93c18dc
  

  Flax Authors

  committed
  93c18dcView commit details

  Copy full SHA for 93c18dc

  Browse repository at this point

• Merge pull request #1622 from 8bitmp3/patch-1

  Show description for d2fcec5
  andsteing

  authored
  d2fcec5View commit details

  Copy full SHA for d2fcec5

  Browse repository at this point

• Add support for transforming methods that are not user functions (e.g.: partial functions).

  Show description for 27b233d
  jheek

  committed
  27b233dView commit details

  Copy full SHA for 27b233d

  Browse repository at this point

• Lint Flax basics

  8bitmp3

  authored
  18454b6View commit details

  Copy full SHA for 18454b6

  Browse repository at this point

Commits on Oct 16, 2021

• Introduce an new system of logical axis names for use with pjit.

  Show description for 11f8151
  

  levskaya

  authored and

  Flax Authors

  committed
  11f8151View commit details

  Copy full SHA for 11f8151

  Browse repository at this point

Commits on Oct 11, 2021

• Merge pull request #1611 from patrickvonplaten:patch-1

  Show description for 8842c69
  

  Flax Authors

  committed
  8842c69View commit details

  Copy full SHA for 8842c69

  Browse repository at this point

• Apply suggestions from code review

  patrickvonplaten

  authored
  8faa5f9View commit details

  Copy full SHA for 8faa5f9

  Browse repository at this point

Commits on Oct 8, 2021

• Add link to Transformers examples

  patrickvonplaten

  authored
  97524c1View commit details

  Copy full SHA for 97524c1

  Browse repository at this point

• Simplify the logic in prefetch_to_device

  Show description for 48b34ab
  

  josipd

  authored and

  Flax Authors

  committed
  48b34abView commit details

  Copy full SHA for 48b34ab

  Browse repository at this point

Commits on Oct 7, 2021

• Add '?' to README image URL to invalidate GitHub caching.

  Show description for 15addcd
  

  ameya98

  authored and

  Flax Authors

  committed
  15addcdView commit details

  Copy full SHA for 15addcd

  Browse repository at this point

• Fix newline in ogbg_molpcba README causing image not to render.

  Show description for 5febb76
  

  ameya98

  authored and

  Flax Authors

  committed
  5febb76View commit details

  Copy full SHA for 5febb76

  Browse repository at this point

Commits on Oct 6, 2021

• Fix RNG splitting in core lift functions for KeyArrays.

  Show description for 6444f97
  

  levskaya

  authored and

  Flax Authors

  committed
  6444f97View commit details

  Copy full SHA for 6444f97

  Browse repository at this point

• Fix a bug in axes_scan for custom prngs.

  Show description for 2aaff69
  

  levskaya

  authored and

  Flax Authors

  committed
  2aaff69View commit details

  Copy full SHA for 2aaff69

  Browse repository at this point

• Implement jax_utils.replicate using jax.device_put_replicated

  Show description for 97dde9a
  

  jheek

  authored and

  Flax Authors

  committed
  97dde9aView commit details

  Copy full SHA for 97dde9a

  Browse repository at this point

Commits on Oct 5, 2021

• Support new JAX custom PRNG key system.

  Show description for a7d5507
  

  levskaya

  authored and

  Flax Authors

  committed
  a7d5507View commit details

  Copy full SHA for a7d5507

  Browse repository at this point

Commits on Oct 4, 2021

• Merge pull request #1599 from google/revert-1597-test_399373571

  Show description for e63b8bd
  marcvanzee

  authored
  e63b8bdView commit details

  Copy full SHA for e63b8bd

  Browse repository at this point

• Revert "Two small fixes for Flax examples"

  marcvanzee

  authored
  595b17dView commit details

  Copy full SHA for 595b17d

  Browse repository at this point

• Merge pull request #1582 from avital/fix-bullet-points

  Show description for eea6045
  avital

  authored
  eea6045View commit details

  Copy full SHA for eea6045

  Browse repository at this point

• Merge pull request #1597 from google/test_399373571

  Show description for ebaf281
  marcvanzee

  authored
  ebaf281View commit details

  Copy full SHA for ebaf281

  Browse repository at this point

• Implements a modular and easily extensible evaluation framework for both TFLite and TFjs. The evaluation framework has the following features:

  Show description for 1820750
  

  marcvanzee

  authored and

  Flax Authors

  committed
  1820750View commit details

  Copy full SHA for 1820750

  Browse repository at this point

• Merge pull request #1595 from dan-zheng:fix-typos

  Show description for 68caa8a
  

  Flax Authors

  committed
  68caa8aView commit details

  Copy full SHA for 68caa8a

  Browse repository at this point

• [JAX] Replace uses of deprecated `jax.ops.index_update(x, idx, y)` APIs with their up-to-date, more succinct equivalent `x.at[idx].set(y)`.

  Show description for facbc5d
  

  hawkinsp

  authored and

  Flax Authors

  committed
  facbc5dView commit details

  Copy full SHA for facbc5d

  Browse repository at this point

Commits on Oct 3, 2021

• Fix typos.

  Show description for 86372e3
  dan-zheng

  committed
  86372e3View commit details

  Copy full SHA for 86372e3

  Browse repository at this point

Commits on Oct 1, 2021

• [JAX] Update JAX users in preparation for a change that makes iteration over a JAX array return JAX arrays, instead of NumPy arrays.

  Show description for b391239
  

  hawkinsp

  authored and

  Flax Authors

  committed
  b391239View commit details

  Copy full SHA for b391239

  Browse repository at this point

Commits on Sep 30, 2021

• Merge pull request #1581 from jheek:map-variables-tranfsorm

  Show description for cba7092
  

  Flax Authors

  committed
  cba7092View commit details

  Copy full SHA for cba7092

  Browse repository at this point

• Map variables for linen

  Show description for 45a3f08
  jheek

  committed
  45a3f08View commit details

  Copy full SHA for 45a3f08

  Browse repository at this point