Alternatives and similar repositories for DiffRast:

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

• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆59Updated 6 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆74Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• AllanYangZhou / universal_neural_functional
  ☆48Updated 11 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated 3 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆33Updated 6 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆33Updated 2 weeks ago
• mag- / gpu_benchmark
  Gpu benchmark
  ☆51Updated this week
• CLAIRE-Labo / flash_attention
  A basic pure pytorch implementation of flash attention
  ☆16Updated 3 months ago
• Silent-Zebra / twisted-smc-lm
  ☆24Updated this week
• vvvm23 / mezo-jax
  JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"
  ☆19Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• DAIOS-AI / mindscript
  A programming language for formal/informal computation.
  ☆41Updated 7 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆27Updated last month
• kklemon / FlashPerceiver
  Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
  ☆24Updated 2 months ago
• riverstone496 / awesome-second-order-optimization
  ☆25Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆45Updated 2 months ago
• hyhieu / easy_pybind
  ☆32Updated 7 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆136Updated 6 months ago
• cloneofsimo / min-fsdp
  ☆75Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆129Updated 2 months ago
• apple / ml-ademamix
  ☆53Updated 2 months ago
• packquickly / schedule_free_optx
  Schedule free optimiser implemented in JAX using Optimistix
  ☆14Updated 8 months ago
• google-deepmind / spectral_ssm
  ☆31Updated 9 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 6 months ago
• iree-org / iree-jax
  ☆49Updated 5 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆85Updated 11 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆30Updated 6 months ago
• indri-voice / vit.triton
  VIT inference in triton because, why not?
  ☆22Updated 8 months ago