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)
  ☆189Updated 3 years ago
• openai / ai-and-efficiency
  Submissions for AI and Efficiency SOTA's
  ☆56Updated 5 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 4 months ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 3 years ago
• openai / LHOPT
  Learned Hyperparameter Optimizers
  ☆59Updated 4 years ago
• soumik12345 / nerf.jax
  A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.
  ☆13Updated 3 years ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated 2 years ago
• huggingface / gaia
  Hugging Face and Pyserini interoperability
  ☆19Updated 2 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated 2 years ago
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 3 years ago
• NolanoOrg / llama-int4-quant
  ☆26Updated 2 years ago
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 3 years ago
• JordanMicahBennett / Supermathematics-and-Artificial-General-Intelligence
  I clearly unravel how I came to invent the supermanifold hypothesis in deep learning, (a part of a system called 'thought curvature') in …
  ☆20Updated 2 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 3 years ago
• google / init2winit
  ☆79Updated this week
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• tensorpro / tpu_rwkv
  JAX implementations of RWKV
  ☆19Updated 2 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, …
  ☆43Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆62Updated 3 years ago
• shawwn / ml-notes
  ☆40Updated 2 years ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated last year
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• 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…
  ☆55Updated 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
• NVIDIA / DALI_deps
  3rd party dependencies for DALI project
  ☆10Updated this week
• google-deepmind / understanding_deep_generative_models_with_generalized_empirical_likelihood
  ☆34Updated 2 years ago
• gpoesia / peano
  An environment for learning formal mathematical reasoning from scratch
  ☆70Updated last year
• prateeky2806 / ComPEFT
  ☆26Updated 2 years ago
• huggingface / optimum-graphcore
  Blazing fast training of 🤗 Transformers on Graphcore IPUs
  ☆85Updated last year
• lvwerra / deep-math
  Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"
  ☆30Updated 2 years ago