Alternatives and similar repositories for jax-enhance

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

• vishwa91 / wire
  wavelet implicit neural representations
  ☆161Updated last year
• lucidrains / x-unet
  Implementation of a U-net complete with efficient attention as well as the latest research findings
  ☆282Updated last year
• balbasty / torch-interpol
  High-order spline interpolation in PyTorch
  ☆67Updated 8 months ago
• samuelepapa / fit-a-nef
  🚀 A powerful library for efficient training of Neural Fields at scale.
  ☆29Updated last year
• kklemon / gon-pytorch
  ☆16Updated 3 years ago
• TheStageAI / TorchIntegral
  Integral Neural Networks in PyTorch
  ☆125Updated 6 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆246Updated 3 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 10 months ago
• lucidrains / uformer-pytorch
  Implementation of Uformer, Attention-based Unet, in Pytorch
  ☆95Updated 3 years ago
• ndahlquist / pytorch-fourier-feature-networks
  Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
  ☆47Updated 4 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 7 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 8 months ago
• NVlabs / RED-diff
  ☆89Updated last year
• francois-rozet / torchist
  NumPy-style histograms in PyTorch
  ☆54Updated last year
• pvjosue / pytorch_convNd
  Functional N-dimensional convolution in Pytorch, recursively calling convNd until reaching conv3d.
  ☆79Updated last month
• annegnx / PnP-Flow
  Code for PnP-Flow: Plug-and-Play image restoration with Flow Matching (ICLR 2025)
  ☆82Updated 3 months ago
• rjbruin / flexconv
  Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…
  ☆115Updated 2 years ago
• v0lta / Jax-Wavelet-Toolbox
  Differentiable and gpu enabled fast wavelet transforms in JAX.
  ☆41Updated 11 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• matthieutrs / ram
  Official implementation of the Reconstruct Anything Model (RAM)
  ☆37Updated 3 weeks ago
• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆36Updated last month
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆106Updated 6 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆53Updated last month
• google-research / spherical-cnn
  ☆117Updated 6 months ago
• jmclong / random-fourier-features-pytorch
  Implementation of "Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains" by Tancik et al.
  ☆93Updated 2 years ago
• boschresearch / multiplicative-filter-networks
  Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.
  ☆95Updated 4 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• vislearn / FFF
  Free-form flows are a generative model training a pair of neural networks via maximum likelihood
  ☆45Updated this week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 5 months ago
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆29Updated 3 months ago