Alternatives and similar repositories for doxa

Users that are interested in doxa are comparing it to the libraries listed below

• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 6 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆203Updated 4 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆242Updated last year
• lucidrains / En-transformer
  Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network
  ☆224Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆346Updated this week
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆254Updated 3 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated 11 months ago
• lucidrains / equiformer-pytorch
  Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…
  ☆270Updated 6 months ago
• lucidrains / Mega-pytorch
  Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena
  ☆204Updated last year
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆305Updated this week
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆46Updated 2 years ago
• recursionpharma / gflownet
  GFlowNet library specialized for graph & molecular data
  ☆253Updated 3 weeks ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated last month
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆115Updated last year
• alexhernandezgarcia / gflownet
  Generative Flow Networks - GFlowNet
  ☆255Updated this week
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆270Updated last week
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆298Updated last year
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆11Updated last year
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆208Updated 5 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• noegroup / bgflow
  Boltzmann Generators and Normalizing Flows in PyTorch
  ☆167Updated last year
• lindermanlab / S5
  ☆292Updated 5 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 months ago
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆56Updated 2 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆342Updated 2 months ago
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆268Updated 2 years ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆252Updated 3 months ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 3 years ago