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 / LHOPT
  Learned Hyperparameter Optimizers
  ☆59Updated 4 years ago
• EleutherAI / pyfra
  Python Research Framework
  ☆106Updated 2 years ago
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 2 years ago
• openai / ai-and-efficiency
  Submissions for AI and Efficiency SOTA's
  ☆56Updated 5 years ago
• frankxu2004 / gpt-neox
  An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.
  ☆17Updated 3 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
• huggingface / gaia
  Hugging Face and Pyserini interoperability
  ☆20Updated 2 years ago
• shawwn / ml-notes
  ☆39Updated 2 years ago
• lvwerra / deep-math
  Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"
  ☆30Updated 2 years ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated 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, …
  ☆40Updated last year
• kyegomez / autogpt-tot
  Simple Autogpt with tree of thoughts
  ☆14Updated 2 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 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
• borisdayma / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆59Updated 3 years ago
• EleutherAI / best-download
  URL downloader supporting checkpointing and continuous checksumming.
  ☆19Updated last year
• aws-neuron / aws-neuron-reference-for-megatron-lm
  ☆14Updated 2 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
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆206Updated last year
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated 2 years ago
• tensorpro / tpu_rwkv
  JAX implementations of RWKV
  ☆19Updated last year
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• moirage / alignment-research-dataset
  A dataset of alignment research and code to reproduce it
  ☆77Updated 2 years ago
• langchain-ai / multi-modal-code-agent
  ☆15Updated last year
• NVIDIA / DALI_deps
  3rd party dependencies for DALI project
  ☆10Updated this week