Alternatives and similar repositories for scikit-jax

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

• watcl-lab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 3 months ago
• metagene-ai / metagene-pretrain
  Pretraining Code for METAGENE-1
  ☆68Updated 8 months ago
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated 10 months ago
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆35Updated last year
• 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…
  ☆54Updated 2 years ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated 2 years ago
• okarthikb / state-space-models
  ☆28Updated last year
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆70Updated 3 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated last year
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated 9 months ago
• jmtomczak / vae_kan_example
  A simple example of VAEs with KANs
  ☆12Updated last year
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 5 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 7 months ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆36Updated 11 months ago
• kyegomez / Pairformer
  Implementation of the Pairformer model used in AlphaFold 3
  ☆14Updated last week
• NousResearch / StripedHyenaTrainer
  ☆62Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• Future-House / ldp
  Framework enabling modular interchange of language agents, environments, and optimizers
  ☆106Updated this week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆101Updated 8 months ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆96Updated 6 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 7 months ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆33Updated 2 years ago
• yobibyte / compressor
  Because we don't have enough time to read everything
  ☆89Updated 11 months ago
• epfml / DenseFormer
  ☆82Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆71Updated 2 months ago
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 3 months ago
• domluna / pubmedFastRAG
  ☆30Updated 7 months ago
• njwfish / DistributionEmbeddings
  ☆28Updated last month
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year