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
  ☆113Updated 8 months ago
• google / grain
  Library for reading and processing ML training data.
  ☆519Updated this week
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆300Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆249Updated 2 years ago
• google-deepmind / enn
  ☆312Updated 5 months ago
• GFNOrg / gflownet
  Generative Flow Networks
  ☆649Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆331Updated 2 years ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆355Updated 2 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…
  ☆273Updated 8 months ago
• lucidrains / En-transformer
  Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network
  ☆226Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆400Updated 4 months ago
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆286Updated last week
• recursionpharma / gflownet
  GFlowNet library specialized for graph & molecular data
  ☆266Updated 2 months ago
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆257Updated last year
• google-deepmind / chex
  ☆880Updated this week
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆369Updated 2 years ago
• google-deepmind / distrax
  ☆589Updated last week
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆473Updated 3 years ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆926Updated last year
• google / learned_optimization
  ☆783Updated this week
• metaopt / torchopt
  TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
  ☆612Updated 3 weeks ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆258Updated 5 months ago
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆273Updated 2 years ago
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,516Updated 4 months ago
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆347Updated 4 months ago
• alexhernandezgarcia / gflownet
  Generative Flow Networks - GFlowNet
  ☆272Updated 2 weeks ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆282Updated last month
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆133Updated last year
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆421Updated 5 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆94Updated 3 months ago