Alternatives and similar repositories for doxa:

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

• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆259Updated this week
• alexhernandezgarcia / gflownet
  Generative Flow Networks - GFlowNet
  ☆239Updated this week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆90Updated 3 months ago
• google-deepmind / tf2jax
  ☆113Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆187Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆335Updated 3 weeks 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…
  ☆263Updated 3 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆193Updated 2 months ago
• recursionpharma / gflownet
  GFlowNet library specialized for graph & molecular data
  ☆234Updated 3 weeks ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆239Updated last year
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆205Updated 11 months ago
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆171Updated 8 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆223Updated last month
• google-deepmind / jaxline
  ☆154Updated last year
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆249Updated last week
• rasmusbergpalm / vnca
  ☆50Updated 2 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• lucidrains / En-transformer
  Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network
  ☆219Updated 10 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 8 months ago
• irhum / esmjax
  ESM2 protein language models in JAX/Flax
  ☆16Updated 2 years ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆232Updated last month
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆265Updated this week
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆305Updated 3 weeks ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated last year
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆49Updated last month
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆47Updated last year
• arnowaczynski / utttai
  AlphaZero-like AI solution for playing Ultimate Tic-Tac-Toe in the browser
  ☆66Updated 2 years ago
• senya-ashukha / simple-equivariant-gnn
  A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks
  ☆130Updated 3 years ago