Alternatives and similar repositories for DiffRast

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

• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆464Updated 9 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆352Updated 2 months ago
• graphcore-research / pytorch-tensor-tracker
  Flexibly track outputs and grad-outputs of torch.nn.Module.
  ☆13Updated 2 years ago
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated last year
• facebookresearch / capi
  Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"
  ☆130Updated 3 weeks ago
• riverstone496 / awesome-second-order-optimization
  ☆28Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆115Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆103Updated this week
• TheStageAI / TorchIntegral
  Integral Neural Networks in PyTorch
  ☆127Updated last year
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆112Updated 7 months ago
• kvfrans / splus
  ☆123Updated 7 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆306Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• hyhieu / easy_pybind
  ☆32Updated last year
• zaydzuhri / softpick-attention
  Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"
  ☆86Updated 4 months ago
• iree-org / iree-jax
  ☆55Updated last year
• AllanYangZhou / universal_neural_functional
  ☆55Updated last year
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆63Updated 2 years ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆243Updated 2 months ago
• Francesco215 / autoregressive_diffusion
  Video Diffusion Model. Autoregressive, long context, efficient training and inference. WIP
  ☆34Updated 5 months ago
• iliao2345 / CompressARC
  ☆214Updated last month
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆138Updated last month
• ESHyperscale / HyperscaleES
  Jax Codebase for Evolutionary Strategies at the Hyperscale
  ☆216Updated last month
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆107Updated 2 months ago
• lucidrains / blackbox-gradient-sensing
  Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…
  ☆20Updated 6 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆82Updated 2 years ago
• geohot / tt-twitch
  tenstorrent kernel from twitch
  ☆28Updated last year
• moritztng / grayskull-attention
  Attention in SRAM on Tenstorrent Grayskull
  ☆40Updated last year
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆495Updated last year