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
• 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
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 6 months ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 3 years ago
• shawwn / ml-notes
  ☆40Updated 3 years ago
• huggingface / gaia
  Hugging Face and Pyserini interoperability
  ☆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
• EleutherAI / pyfra
  Python Research Framework
  ☆107Updated 3 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 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
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 3 years ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 4 years ago
• huggingface / optimum-graphcore
  Blazing fast training of 🤗 Transformers on Graphcore IPUs
  ☆86Updated last year
• attentionneuron / attentionneuron.github.io
  ☆26Updated 3 years ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆46Updated 2 years ago
• EleutherAI / best-download
  URL downloader supporting checkpointing and continuous checksumming.
  ☆19Updated 2 years ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated 2 years ago
• tensorpro / tpu_rwkv
  JAX implementations of RWKV
  ☆19Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆63Updated 3 years ago
• princeton-nlp / DataMUX
  [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks
  ☆60Updated 3 years ago
• NolanoOrg / llama-int4-quant
  ☆26Updated 2 years ago
• google-research / precondition
  ☆31Updated last month
• 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 2 years ago
• chai-research / lmgym
  Code base for internal reward models and PPO training
  ☆24Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year