jax

Hello,

The code says that it will add compatibility for Postponed Evaluation of Annotations (PEP 563) when Python 3.9 is released (which already happened on 2020.10.5). Is there any plan to complete this?

https://github.com/huggingface/transformers/blob/2c2a31ffbcfe03339b1721348781aac4fc05bc5e/src/transformers/hf_argparser.py#L85-L90

>>> lax.broadcast_shapes((-1,), (1,))                                                                                                                    
(-1,)
>>> np.broadcast_shapes((-1,), (1,))                                                                                                                     
------------------------------------------------------------------------

Is there a way to train a bidirectional RNN (like LSTM or GRU) on trax nowadays?

Love the book! Ran into this display issue when viewing it on Chrome's PDF viewer:

This will help some forum users avoid issues like using HalfNormal(loc, scale) for HalfNormal distribution. In such case, validate_args will take the value of scale (because HalfNormal does not have loc parameter) and raise some misleading errors.

Dear Brax team,

Since Brax is fully differentiable, I thought it'd be possible to use it like DiffTaichi or GradSim for system identification (e.g. determining the mass of an object from a trajectory and known force) but I couldn't find any example for this.
Do you happen to have any demo or tips for this?

From the top of my head I would do something like this:
Let's say the task is to es

jit and unroll don't seem to be documented in https://jaxopt.github.io/stable/_autosummary/jaxopt.GradientDescent.html#jaxopt-gradientdescent

jax

Here are 269 public repositories matching this topic...

huggingface / transformers

google / jax

google / trax

arogozhnikov / einops

probml / pyprobml

BoltzmannEntropy / interviews.ai

tensorflow / datasets

google / flax

explosion / thinc

bethgelab / foolbox

deepmind / dm-haiku

google / neural-tangents

pyro-ppl / numpyro

google-research / big_transfer

tensorly / tensorly

google / brax

lululxvi / deepxde

deepmind / jraph

google-research / scenic

parrt / tensor-sensor

jonasrauber / eagerpy

unifyai / ivy

n2cholas / awesome-jax

yang-song / score_sde

google / jaxopt

google-research / long-range-arena

google-research / fast-soft-sort

patrick-kidger / diffrax

poets-ai / elegy

patrick-kidger / equinox

Improve this page

Add this topic to your repo