165 Open source projects that are alternatives of or similar to jax-models

Chef Transformer 🍲 .

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022

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.

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

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

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

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

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

KoCLIP: Korean port of OpenAI CLIP, in Flax

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

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

JAX implementations of core Deep RL algorithms

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

[NeurIPS 2021] COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretraining

Hashformers is a framework for hashtag segmentation with transformers.

Building a model to recognize incentives for landscape restoration in environmental policies from Latin America, the US and India. Bringing NLP to the world of policy analysis through an extensible framework that includes scraping, preprocessing, active learning and text analysis pipelines.

[AAAI 2022] This is the official PyTorch implementation of "Less is More: Pay Less Attention in Vision Transformers"

Must-read papers on Natural Language Processing (NLP)

Transformer based Bangla Speech Recognition

Emotion recognition in conversation

Tutorials on getting started with PyTorch and TorchText for sentiment analysis.

Bangla-Bert is a pretrained bert model for Bengali language

A library for finding knowledge neurons in pretrained transformer models.

ADAM implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.

Collection of NLP model explanations and accompanying analysis tools

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

[ICCV 2021] Official implementation of "Scalable Vision Transformers with Hierarchical Pooling"

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

KoBERT on 🤗 Huggingface Transformers 🤗 (with Bug Fixed)

Zero-Shot Text-to-Image Generation VQGAN+CLIP Dockerized

KB국민은행에서 제공하는 경제/금융 도메인에 특화된 한국어 ALBERT 모델

Yuno is context based search engine for anime.

Graphsignal Python agent

Implementation of TransGanFormer, an all-attention GAN that combines the finding from the recent GanFormer and TransGan paper

Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

A model library for exploring state-of-the-art deep learning topologies and techniques for optimizing Natural Language Processing neural networks

State-of-the-art NLP through transformer models in a modular design and consistent APIs.

Simple transformer implementations that I can understand

🧑‍🏫 50! Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

🌏 A zero-dependency networking solution for building modern and secure iOS, watchOS, macOS and tvOS applications.

Computational Fluid Dynamics in JAX

Transformers for Classification, NER, QA, Language Modelling, Language Generation, T5, Multi-Modal, and Conversational AI

An Open-Source Package for Chinese Open-domain Conversational Chatbot (中文闲聊对话系统，一键部署微信闲聊机器人)

classy is a simple-to-use library for building high-performance Machine Learning models in NLP.

General Multi-label Image Classification with Transformers

(TPAMI 2022) Snowflake Point Deconvolution for Point Cloud Completion and Generation with Skip-Transformer

State of the Art Natural Language Processing

Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch

Text2Text: Cross-lingual natural language processing and generation toolkit

TransmogrifAI (pronounced trăns-mŏgˈrə-fī) is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning

Load What You Need: Smaller Multilingual Transformers for Pytorch and TensorFlow 2.0.

Deploy transformers serverless on AWS Lambda

Implementation of STAM (Space Time Attention Model), a pure and simple attention model that reaches SOTA for video classification

Powerful unsupervised domain adaptation method for dense retrieval. Requires only unlabeled corpus and yields massive improvement: "GPL: Generative Pseudo Labeling for Unsupervised Domain Adaptation of Dense Retrieval" https://arxiv.org/abs/2112.07577

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Multi-class text categorization using state-of-the-art pre-trained contextualized language models, e.g. BERT

Text Classification in Turkish Texts with Bert

Super easy library for BERT based NLP models

Utility for analyzing Transformer based representations of language.