54 results found Sort:
- Filter by Primary Language:
- Python (36)
- Jupyter Notebook (13)
- C# (2)
- C++ (2)
- CSS (1)
- +
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Created
2018-10-29
17,407 commits to main branch, last one 15 hours ago
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Created
2022-05-30
4,731 commits to main branch, last one 23 hours ago
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
Flax Engine – multi-platform 3D game engine
Created
2020-12-09
10,386 commits to master branch, last one 18 hours 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 24 days ago
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 3 months ago
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
A Library for Uncertainty Quantification.
Created
2022-11-17
235 commits to main branch, last one 2 months 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 6 months ago
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
Tevatron - A flexible toolkit for neural retrieval research and development.
Created
2021-09-09
280 commits to main branch, last one about a month ago
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
CLU lets you write beautiful training loops in JAX.
Created
2020-09-11
284 commits to main branch, last one 3 months ago
Orbax provides common checkpointing and persistence utilities for JAX users
Created
2022-05-06
1,058 commits to main branch, last one a day ago
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
Accelerate, Optimize performance with streamlined training and serving options with JAX.
Created
2023-04-09
1,878 commits to main branch, last one a day ago
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
general purpose programming language, in the vein of C++
Created
2014-11-14
1,960 commits to master branch, last one 3 years ago
KoCLIP: Korean port of OpenAI CLIP, in Flax
Created
2021-07-03
30 commits to master branch, last one about a year ago
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
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
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
Collection of example projects for Flax Engine
Created
2018-01-25
107 commits to master branch, last one about a month 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
Flax Engine Documentation
Created
2017-03-28
1,429 commits to master branch, last one 16 days ago
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
Created
2022-07-21
10 commits to master branch, last one 8 months ago