Alternatives and similar repositories for jax-enhance

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

• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆47Updated last month
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆28Updated 4 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
• google-research / spherical-cnn
  ☆118Updated 6 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• mauriceweiler / MobiusCNNs
  This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…
  ☆72Updated last year
• vishwa91 / wire
  wavelet implicit neural representations
  ☆163Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 6 months ago
• pvjosue / pytorch_convNd
  Functional N-dimensional convolution in Pytorch, recursively calling convNd until reaching conv3d.
  ☆79Updated last month
• ermongroup / self-similarity-prior
  Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations
  ☆28Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 5 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated 11 months ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month
• ndahlquist / pytorch-fourier-feature-networks
  Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
  ☆48Updated 5 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• jloveric / high-order-layers-torch
  High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…
  ☆45Updated last year
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 9 months ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆43Updated 5 years ago
• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆36Updated 2 months ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 3 years ago
• Ujjawal-K-Panchal / coshnet
  ☆51Updated last year
• rjbruin / flexconv
  Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…
  ☆116Updated 2 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 8 months ago
• ehoogeboom / convolution_exponential_and_sylvester
  ☆28Updated 3 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 11 months ago
• vislearn / FFF
  Free-form flows are a generative model training a pair of neural networks via maximum likelihood
  ☆45Updated last week
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆60Updated last year
• TheStageAI / TorchIntegral
  Integral Neural Networks in PyTorch
  ☆125Updated 6 months ago