#

flax

Here are 90 public repositories matching this topic...

transformers

huggingface / transformers

Open

Add missing tokenizer test files [still need 3 first time contributors]

24

SaulLu commented Apr 6, 2022

🚀 Add missing tokenizer test files

Several tokenizers currently have no associated tests. I think that adding the test file for one of these tokenizers could be a very good way to make a first contribution to transformers.

Tokenizers concerned

not yet claimed

LED
RemBert
RetriBert

claimed

Flaubert @anmolsjoshi
Longformer @tgadeliya
[

Read more

Good First Issue

Open

ONNXConfig: Add a configuration for all available models

19

Open

[Community Event] Doc Tests Sprint

73

Find more good first issues

probml / pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy

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

Updated Apr 27, 2022
Jupyter Notebook

FlaxEngine

FlaxEngine / FlaxEngine

Flax Engine – multi-platform 3D game engine

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

Updated Apr 24, 2022
C++

yang-song / score_sde

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

Long Range Arena for Benchmarking Efficient Transformers

nlp deep-learning transformers attention flax jax

Updated Apr 20, 2022
Python

ikostrikov / jaxrl

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 Mar 27, 2022
Jupyter Notebook

get-started-with-JAX

gordicaleksa / get-started-with-JAX

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

python machine-learning tutorial deep-learning jupyter numpy flax haiku lax jax xla optax learn-jax

Updated Jan 22, 2022
Jupyter Notebook

flax-lang / flax

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

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#

luyug / GradCache

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

deep-learning pytorch flax jax contrastive-learning

Updated Jan 21, 2022
Python

google-research / prompt-tuning

Original Implementation of Prompt Tuning from Lester, et al, 2021

nlp machine-learning flax language-model jax prompt-tuning

Updated Apr 26, 2022
Python

matthias-wright / flaxmodels

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

vgg resnet flax pretrained-models jax gpt2 stylegan2

Updated Jan 10, 2022
Python

texttron / tevatron

Tevatron - A flexible toolkit for dense retrieval research and development.

information-retrieval pytorch transformer question-answering flax dpr jax dense-retrieval

Updated Apr 15, 2022
Python

rwightman / efficientnet-jax

Sponsor

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

google / CommonLoopUtils

CLU lets you write beautiful training loops in JAX.

Updated Apr 26, 2022
Jupyter Notebook

jaketae / koclip

KoCLIP: Korean port of OpenAI CLIP, in Flax

flax jax roberta vision-transformer openai-clip

Updated Aug 16, 2021
Jupyter Notebook

n2cholas / jax-resnet

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 Dec 10, 2021
Python

DarshanDeshpande / jax-models

Unofficial JAX implementations of deep learning research papers

machine-learning computer-vision deep-learning transformers pytorch artificial-intelligence convolutional-neural-networks flax jax research-paper-implementation

Updated Feb 28, 2022
Python

yang-song / score_flow

Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)

flax normalizing-flows jax diffusion-models score-matching neural-odes score-based-generative-modeling neurips-2021 score-based-generative-models

Updated Dec 9, 2021
Python

FlaxDocs

FlaxEngine / FlaxDocs

Flax Engine Documentation

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

Updated Apr 14, 2022
CSS

henry-prior / jax-rl

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 Mar 15, 2022
Python

FlaxSamples

FlaxEngine / FlaxSamples

Collection of example projects for Flax Engine

gamedev visual-studio templates samples flax flax-engine

Updated Mar 2, 2022
C#

evgenii-nikishin / omd

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

revsic / jax-variational-diffwave

Jax/Flax implementation of Variational-DiffWave.

vdm flax jax diffwave variational-diffusion-models

Updated Feb 27, 2022
Python

nestordemeure / flaxOptimizers

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

flax optimizers

Updated Aug 6, 2021
Python

chef-transformer

chef-transformer / chef-transformer

Chef Transformer 🍲 .

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

Updated Mar 28, 2022
Python

nipunsadvilkar / roberta-base-mr

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

transformers flax marathi jax roberta huggingface

Updated Jul 18, 2021
Python

myagues / flax_nerf

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

gcucurull / maml_flax

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

flax maml meta-learning jax

Updated Sep 18, 2020
Python

lweitkamp / GANs-JAX

Implementation of several Generative Adversarial Networks in JAX / Flax

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

Updated Apr 14, 2022
Jupyter Notebook

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."