Alternatives and similar repositories for scikit-jax:

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

• SamsungSAILMontreal / AnyMolGenCritic
  ☆22Updated 6 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 9 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 10 months ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated last year
• jfpuget / ARC-AGI-Challenge-2024
  ☆47Updated 4 months ago
• okarthikb / state-space-models
  ☆27Updated 8 months ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆33Updated 6 months ago
• patrick-kidger / esm2quinox
  An implementation of ESM2 in Equinox+JAX
  ☆25Updated last month
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 2 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 5 months ago
• Future-House / ldp
  Agent framework for constructing language model agents and training on constructive tasks.
  ☆68Updated last week
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 9 months 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…
  ☆52Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆76Updated 3 weeks ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 8 months ago
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆63Updated last month
• xjdr-alt / muzero_sketch
  ☆38Updated 8 months ago
• cambridge-mlg / jolt
  ☆11Updated last month
• epfml / DenseFormer
  ☆79Updated 11 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆32Updated last year
• ltgoslo / bert-in-context
  Official implementation of "BERTs are Generative In-Context Learners"
  ☆26Updated 2 weeks ago
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆34Updated 10 months ago
• domluna / pubmedFastRAG
  ☆28Updated last month
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆23Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆42Updated last week
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆62Updated 2 weeks ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year