Alternatives and similar repositories for PaLM-flax

Users that are interested in PaLM-flax are comparing it to the libraries listed below

• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 3 years ago
• openai / ai-and-efficiency
  Submissions for AI and Efficiency SOTA's
  ☆56Updated 5 years ago
• openai / LHOPT
  Learned Hyperparameter Optimizers
  ☆59Updated 4 years ago
• huggingface / gaia
  Hugging Face and Pyserini interoperability
  ☆20Updated 2 years ago
• EshaanAgg / Research-Literature-Manager
  A simple way to manage and store the data related to all your research papers!
  ☆18Updated 2 years ago
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆71Updated 2 weeks ago
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 2 years ago
• Quansight-Labs / numpy_pytorch_interop
  ☆20Updated last year
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆24Updated last year
• google / init2winit
  ☆76Updated last week
• kyegomez / autogpt-tot
  Simple Autogpt with tree of thoughts
  ☆14Updated 2 years ago
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• VITA-Group / ChainCoder
  [ICML 2023] "Outline, Then Details: Syntactically Guided Coarse-To-Fine Code Generation", Wenqing Zheng, S P Sharan, Ajay Kumar Jaiswal, …
  ☆40Updated last year
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• lucidrains / esbn-transformer
  An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols
  ☆16Updated 3 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆204Updated last year
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• google-research / structured-additive-IR
  ☆61Updated last week
• google-deepmind / understanding_deep_generative_models_with_generalized_empirical_likelihood
  ☆34Updated 2 years ago
• TheoCoombes / crawlingathome
  A client library for LAION's effort to filter CommonCrawl with CLIP, building a large scale image-text dataset.
  ☆32Updated 2 years ago
• microsoft / ARXGEN
  Scripts to parse arxiv documents for NLP tasks
  ☆18Updated 2 years ago
• xrsrke / stable-diffusion-from-scratch
  Implementation of Stable Diffusion from scratch [WORK IN PROGRESS]
  ☆22Updated 2 years ago
• aws-neuron / aws-neuron-reference-for-megatron-lm
  ☆14Updated last year
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• lucidrains / flash-genomics-model
  My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…
  ☆53Updated 2 years ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated 2 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year