Alternatives and similar repositories for scikit-jax

Users that are interested in scikit-jax are comparing it to the libraries listed below

• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated 2 years ago
• patrick-kidger / esm2quinox
  An implementation of ESM2 in Equinox+JAX
  ☆26Updated 3 months ago
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 2 months ago
• cambridge-mlg / jolt
  ☆11Updated 3 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 11 months ago
• okarthikb / state-space-models
  ☆27Updated 10 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated last week
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆33Updated 8 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 4 months ago
• kyegomez / Pairformer
  Implementation of the Pairformer model used in AlphaFold 3
  ☆12Updated this week
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆62Updated last week
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆35Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 11 months ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated 2 years ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated 11 months ago
• lucidrains / hyena-dna
  Fork of HyenaDNA, a long-range genomic foundation model built with Hyena
  ☆9Updated last year
• FLAIROx / cultural-accumulation
  ☆13Updated 10 months ago
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆15Updated 4 months ago
• jmtomczak / vae_kan_example
  A simple example of VAEs with KANs
  ☆12Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated last week
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 7 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 8 months ago
• instadeepai / metalic
  Meta in-context learning for protein fitness prediction
  ☆11Updated 3 months ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 4 months ago
• etimush / ARC_NCA
  Repo for solving arc problems with an Neural Cellular Automata
  ☆15Updated 2 weeks ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• Artur-Galstyan / statedict2pytree
  ☆43Updated this week
• Future-House / ldp
  A Software Framework Enabling Modular Interchange of Language Agents, Environments, and Optimizers
  ☆87Updated last week