Alternatives and similar repositories for scikit-jax:

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

• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated last month
• jmtomczak / vae_kan_example
  A simple example of VAEs with KANs
  ☆12Updated 9 months ago
• okarthikb / state-space-models
  ☆27Updated 7 months ago
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 5 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 7 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 8 months ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆32Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 4 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 8 months ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆47Updated 3 months ago
• pgniewko / forward_forward_vhts
  The Forward-Forward Algorithm for Drug Discovery
  ☆34Updated 2 years ago
• Future-House / ldp
  Agent framework for constructing language model agents and training on constructive tasks.
  ☆65Updated this week
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆33Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• kyegomez / Pairformer
  Implementation of the Pairformer model used in AlphaFold 3
  ☆12Updated last week
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆34Updated 9 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 9 months ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated 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…
  ☆52Updated last year
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆32Updated 5 months ago
• requeima / llm_processes
  ☆44Updated 2 weeks ago
• krypticmouse / matryoshka-representation-learning
  PyTorch implementation for MRL
  ☆18Updated last year
• christopher-beckham / coms-are-energy-models
  Official code for paper: Conservative objective models are a special kind of contrastive divergence-based energy model
  ☆14Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 5 months ago
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆69Updated last year
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 3 months ago