#

flax

Here are 66 public repositories matching this topic...

transformers

huggingface / transformers

Star

Open

Raise exceptions instead of using assertions for control flow

10

willfrey commented Jul 19, 2021

https://github.com/huggingface/transformers/blob/546dc24e0883e5e9f5eb06ec8060e3e6ccc5f6d7/src/transformers/models/gpt2/modeling_gpt2.py#L698

Assertions can't be relied upon for control flow because they can be disabled, as per the following:

$ python --help
usage: python [option] ... [-c cmd | -m mod | file | -] [arg] ...
...
-O     : remove assert and __debug__-dependent statem

Read more

Good First Issue

Open

[Performance] Tracking open Issues and PRs (pytorch transformers)

Open

Getting time offsets of beginning and end of each word in Wav2Vec2

17

Find more good first issues

probml / pyprobml

Star

Python code for "Machine learning: a probabilistic perspective" (2nd edition)

machine-learning tensorflow pytorch colab pml probabilistic-programming flax jupyter-notebooks pymc3 pyro jax numpyro

Updated Sep 29, 2021
Python

FlaxEngine

FlaxEngine / FlaxEngine

Star

Flax Engine – multi-platform 3D game engine

c-sharp gamedev multi-platform game-engine cpp engine game-development flax flax-engine flaxengine

Updated Sep 29, 2021
C++

yang-song / score_sde

Star

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

flax stochastic-differential-equations inverse-problems generative-models jax diffusion-models score-matching score-based-generative-modeling controllable-generation iclr-2021

Updated Aug 26, 2021
Jupyter Notebook

google-research / long-range-arena

Star

Long Range Arena for Benchmarking Efficient Transformers

nlp deep-learning transformers attention flax jax

Updated Aug 30, 2021
Python

ikostrikov / jaxrl

Star

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

reinforcement-learning deep-learning deep-reinforcement-learning gym flax behavioral-cloning sac continuous-control deep-deterministic-policy-gradient jax soft-actor-critic offline-reinforcement-learning batch-reinforcement-learning

Updated Sep 29, 2021
Jupyter Notebook

flax-lang / flax

Star

general purpose programming language, in the vein of C++

language programming-language c-plus-plus compiler llvm language-design flax

Updated Jun 20, 2021
C++

FlaxAPI

FlaxEngine / FlaxAPI

Star

Old repository with C# Editor and C# API for creating games in Flax Engine

editor api gamedev games game-engine game-development flax 3d-game-engine 3d-engine flax-engine flax-editor

Updated Dec 24, 2020
C#

rwightman / efficientnet-jax

Sponsor Star

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

flax mixnet tpu jax mobilenetv2 mobilenetv3 efficientnet objax flax-linen

Updated May 31, 2021
Python

matthias-wright / flaxmodels

Star

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet.

vgg resnet flax pretrained-models jax gpt2 stylegan2

Updated Sep 24, 2021
Python

google / CommonLoopUtils

Star

CLU lets you write beautiful training loops in JAX.

Updated Sep 24, 2021
Jupyter Notebook

FlaxDocs

FlaxEngine / FlaxDocs

Star

Flax Engine Documentation

gamedev documentation docs game-engine flax docfx flax-engine flax-documentation flaxengine

Updated Sep 29, 2021
CSS

jaketae / koclip

Star

KoCLIP: Korean port of OpenAI CLIP, in Flax

flax jax roberta vision-transformer openai-clip

Updated Aug 16, 2021
Jupyter Notebook

FlaxSamples

FlaxEngine / FlaxSamples

Star

Collection of example projects for Flax Engine

gamedev visual-studio templates samples flax flax-engine

Updated Aug 6, 2021
C#

n2cholas / jax-resnet

Star

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

machine-learning computer-vision deep-learning neural-networks resnet flax jax

Updated Aug 11, 2021
Python

henry-prior / jax-rl

Star

JAX implementations of core Deep RL algorithms

reinforcement-learning deep-learning deep-reinforcement-learning flax deepmind sac actor-critic maximum-a-posteriori-estimation mujoco jax td3 soft-actor-critic

Updated Feb 10, 2021
Python

evgenii-nikishin / omd

Star

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

machine-learning reinforcement-learning deep-learning automatic-differentiation deep-reinforcement-learning dqn gym flax haiku sac model-based-rl jax model-based-reinforcement-learning soft-actor-critic dm-haiku

Updated Jun 14, 2021
Jupyter Notebook

nestordemeure / flaxOptimizers

Star

A collection of optimizers, some arcane others well known, for Flax.

flax optimizers

Updated Aug 6, 2021
Python

nipunsadvilkar / roberta-base-mr

Star

RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week

transformers flax marathi jax roberta huggingface

Updated Jul 18, 2021
Python

chef-transformer

chef-transformer / chef-transformer

Star

Chef Transformer 🍲 .

recipes transformers flax recipe-generator jax huggingface t5 recipe-generation google-t5

Updated Jul 27, 2021
Jupyter Notebook

revsic / jax-variational-diffwave

Star

Jax/Flax implementation of Variational-DiffWave.

vdm flax jax diffwave variational-diffusion-models

Updated Sep 6, 2021
Python

nestordemeure / AdaHessianJax

Star

Jax implementation of the AdaHessian optimizer

deep-learning optimizer flax hessian jax adahessian

Updated Mar 11, 2021
Python

lweitkamp / GANs-JAX

Star

Implementation of several Generative Adversarial Networks in JAX / Flax

python machine-learning generative-adversarial-network flax gans jax

Updated Apr 11, 2021
Jupyter Notebook

myagues / flax_nerf

Star

Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API

colab flax nerf jax neural-radiance-fields

Updated Sep 25, 2021
Jupyter Notebook

sooheon / perceiver-jax

Star

Perceiver (transformer variant) implemented in JAX and Flax

deep-learning transformer flax jax perceiver

Updated Mar 29, 2021
Python

deterministic-algorithms-lab / Jax-Journey

Star

A pathway and collection of resources to learning Jax from beginning to advance.

machine-learning deep-learning flax jax dm-haiku

Updated Jan 2, 2021
Jupyter Notebook

n2cholas / progan-flax

Star

Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation

gan neural-networks flax progressive-gan jax

Updated May 24, 2021
Python

vballoli / vit-flax

Star

Implementation of Vision Transformers in Flax

flax jax vision-transformer

Updated Oct 12, 2020
Python

FlaxCommunityProjects / FlaxCollaborate

Star

collaboration flax flax-engine

Updated May 12, 2021
C#

gcucurull / maml_flax

Star

Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.

flax maml meta-learning jax

Updated Sep 18, 2020
Python

Improve this page

Add a description, image, and links to the flax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."