google-research/spherical-cnn

Differentiable signal processing on the sphere for PyTorch

S2FFT: Differentiable and accelerated spherical transforms

Scenic: A Jax Library for Computer Vision Research and Beyond

E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

SPHARM-Net: Spherical Harmonics-based Convolutional Neural Network

This project is dedicated to the implementation and research of Kolmogorov-Arnold convolutional networks. The repository includes implementations of 1D, 2D, and 3D convolutions with different kernels, ResNet-like and DenseNet-like models, training code based on accelerate/PyTorch, as well as scripts for experiments with CIFAR-10 and Tiny ImageNet.

JAX-based neural network library