dtunai / xLSTM-Jax

Related projects ⓘ

Alternatives and complementary repositories for xLSTM-Jax

• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆30Updated 4 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆59Updated last month
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆39Updated last week
• enajx / HyperNCA
  ☆36Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆102Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆82Updated 2 weeks ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆34Updated last month
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆87Updated last month
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆127Updated 2 weeks ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• google-deepmind / tf2jax
  ☆105Updated 2 weeks ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆77Updated 3 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆13Updated 11 months ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆35Updated 2 months ago
• google-deepmind / spectral_ssm
  ☆28Updated 7 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆28Updated last year
• IBM / abductive-rule-learner-with-context-awareness
  PyTorch Implementation of the paper "Towards Learning Abductive Reasoning using VSA Distributed Representations".
  ☆12Updated 3 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆88Updated last year
• radarFudan / mamba-minimal-jax
  ☆29Updated 2 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated last month
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆17Updated 2 weeks ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago