probabilists / azula

Related projects: ⓘ

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆62Updated last month
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆61Updated 4 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆36Updated last year
• francois-rozet / papers-101
  Implementation of papers in 101 lines of code.
  ☆18Updated 10 months ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆119Updated 11 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆187Updated 10 months ago
• BlackHC / neural_net_checklist
  ☆137Updated last month
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆82Updated 3 weeks ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆52Updated last month
• PolymathicAI / xVal
  Repository for code used in the xVal paper
  ☆110Updated 5 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆85Updated last month
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆66Updated 3 months ago
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆108Updated 3 weeks ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆118Updated 8 months ago
• ethansmith2000 / TransformerExperiments
  ☆18Updated last month
• cgarciae / einop
  ☆56Updated 2 years ago
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆108Updated last year
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆69Updated 7 months ago
• google-research / spherical-cnn
  ☆103Updated last month
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆53Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆66Updated 7 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆76Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 3 weeks ago
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆83Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆91Updated last week
• gle-bellier / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆24Updated 3 weeks ago
• lucidrains / hyena-pytorch
  ☆23Updated this week