Awesome Open Source

Programming Languages

Search results for python flax

50 search results found

Transformers ⭐ 124,049

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Easylm ⭐ 2,087

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Fortuna ⭐ 824

A Library for Uncertainty Quantification.

Long Range Arena ⭐ 635

Long Range Arena for Benchmarking Efficient Transformers

Prompt Tuning ⭐ 580

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

Get Started With Jax ⭐ 487

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.

Tevatron ⭐ 334

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

Gradcache ⭐ 254

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

Flaxmodels ⭐ 201

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

Orbax provides common utility libraries for JAX users.

Train very large language models in Jax.

Efficientnet Jax ⭐ 121

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

KoCLIP: Korean port of OpenAI CLIP, in Flax

EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥

Jax Models ⭐ 82

Unofficial JAX implementations of deep learning research papers

Jestimator ⭐ 77

Amos optimizer with JEstimator lib.

Jax Resnet ⭐ 67

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

M3ae_public ⭐ 58

Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation

Imgx Diffseg ⭐ 53

A JAX-based deep learning framework for image segmentation using diffusion models.

JAX implementations of core Deep RL algorithms

Grad-DFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.

Sac N Jax ⭐ 36

Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch

Score_flow ⭐ 35

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

MLSys Workshop NeurIPS 2023 - Redco: A Lightweight Tool to Automate Distributed Training and Inference

Chef Transformer ⭐ 29

Chef Transformer 🍲 .

Jax Variational Diffwave ⭐ 28

Jax/Flax implementation of Variational-DiffWave.

Train large Huggingface models with flexible data/model parallelism in Flax/JAX

a modular reinforcement learning library with JAX agents

Generative neural networks for 3D terrain.

Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.

FID computation in Jax/Flax.

A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.

Flaxoptimizers ⭐ 13

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

Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`

Gans Jax ⭐ 12

Implementation of several Generative Adversarial Networks in JAX / Flax

Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.

Fjformer ⭐ 10

Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights!

Jax So Vits Svc 5.0 ⭐ 10

A so-vits-svc-5.0 rewritten in jax and flax. Works properly on TPU v2/v3.

Jax_trainer ⭐ 10

Lightning-like training API for JAX with Flax

Probabilistic deep learning using JAX

Roberta Base Mr ⭐ 9

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

Holographic Reduced Representations ⭐ 9

Holographic Reduced Representations

Interoperating between (Deep) Reiforcement Learning libraries

Adahessianjax ⭐ 9

Jax implementation of the AdaHessian optimizer

Functional Diffusion Processes ⭐ 8

Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).

Polyglot Jax Inference ⭐ 8

TPU에서 한국어용 LLM 추론을 위한 Jax/Flax 구현체입니다.

Rl Basics ⭐ 8

Simple single file implementation of Deep Reinforcement Learning algorithms

Dash Docset Jax ⭐ 6

Create nice dash/zeal docset for JAX+Flax

Starcoder Jax ⭐ 6

a Jax/Flax inference code of StarCoder

Jax_examples ⭐ 5

Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)

Related Searches

Python Machine Learning (19,284)

Python Dataset (14,792)

Python Tensorflow (13,736)

Python Deep Learning (13,092)

Python Jupyter Notebook (12,976)

Python Natural Language Processing (9,064)

Python Artificial Intelligence (8,580)

Python Pytorch (7,877)

Python Numpy (5,701)

Python Segmentation (4,571)

1-50 of 50 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.