54 results found Sort:

26.9k
134.5k
apache-2.0
1.1k
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Created 2018-10-29
17,340 commits to main branch, last one a day ago
5.4k
26.1k
apache-2.0
201
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Created 2022-05-30
4,701 commits to main branch, last one a day ago
1.5k
6.5k
mit
192
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Created 2016-08-17
3,036 commits to master branch, last one 3 months ago
566
5.8k
other
116
Flax Engine – multi-platform 3D game engine
Created 2020-12-09
10,356 commits to master branch, last one a day ago
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Created 2020-07-25
570 commits to master branch, last one 10 days ago
255
2.4k
apache-2.0
43
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Created 2022-11-22
229 commits to main branch, last one 2 months ago
205
1.5k
apache-2.0
19
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
Created 2021-01-03
24 commits to main branch, last one 3 years ago
Long Range Arena for Benchmarking Efficient Transformers
Created 2020-11-07
37 commits to main branch, last one 2 years ago
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 t...
Created 2021-10-28
33 commits to main branch, last one 2 years ago
Original Implementation of Prompt Tuning from Lester, et al, 2021
Created 2021-09-07
89 commits to main branch, last one 5 months ago
69
623
mit
12
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Created 2021-01-16
133 commits to main branch, last one 2 years ago
99
515
apache-2.0
12
Tevatron - A flexible toolkit for neural retrieval research and development.
Created 2021-09-09
280 commits to main branch, last one 17 days ago
22
359
apache-2.0
9
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
Created 2021-06-08
35 commits to main branch, last one 7 months ago
31
321
apache-2.0
10
CLU lets you write beautiful training loops in JAX.
Created 2020-09-11
284 commits to main branch, last one 2 months ago
33
296
apache-2.0
10
Orbax provides common checkpointing and persistence utilities for JAX users
Created 2022-05-06
1,020 commits to main branch, last one 8 hours ago
11
274
mit
9
A Jax-based library for designing and training transformer models from scratch.
Created 2023-08-22
158 commits to main branch, last one 2 months ago
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Created 2021-05-16
90 commits to main branch, last one about a year ago
25
201
apache-2.0
13
Accelerate, Optimize performance with streamlined training and serving options with JAX.
Created 2023-04-09
1,850 commits to main branch, last one 2 days ago
17
195
mit
10
Train very large language models in Jax.
Created 2022-10-02
74 commits to main branch, last one about a year ago
Unofficial JAX implementations of deep learning research papers
Created 2022-01-08
30 commits to main branch, last one 2 years ago
10
146
apache-2.0
10
general purpose programming language, in the vein of C++
Created 2014-11-14
1,960 commits to master branch, last one 3 years ago
17
142
apache-2.0
4
KoCLIP: Korean port of OpenAI CLIP, in Flax
Created 2021-07-03
30 commits to master branch, last one about a year ago
25
133
unknown
8
Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
Created 2021-10-18
2 commits to main branch, last one 2 years ago
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Created 2020-09-12
50 commits to master branch, last one 10 months ago
19
123
apache-2.0
3
A multi-modal CLIP model trained on the medical dataset ROCO
Created 2021-07-23
21 commits to master branch, last one 3 months ago
Collection of example projects for Flax Engine
Created 2018-01-25
107 commits to master branch, last one 28 days ago
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Created 2020-12-29
46 commits to main branch, last one 2 years ago
64
101
cc-by-4.0
10
Flax Engine Documentation
Created 2017-03-28
1,429 commits to master branch, last one a day ago
10
100
apache-2.0
6
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
Created 2022-07-21
10 commits to master branch, last one 7 months ago