Alternatives and similar repositories for bagz

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

• google / array_record
  ☆121Updated this week
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆45Updated 2 years ago
• openai / chz
  ☆182Updated 3 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆107Updated 2 weeks ago
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆71Updated 5 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆124Updated last month
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆97Updated this week
• google-deepmind / einshape
  ☆107Updated last year
• njwfish / DistributionEmbeddings
  ☆33Updated last week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated 10 months ago
• hyhieu / easy_pybind
  ☆32Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆91Updated 4 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 3 months ago
• lucidrains / simplicial-attention
  Implementation of 2-simplicial attention proposed by Clift et al. (2019) and the recent attempt to make practical in Fast and Simplex, Ro…
  ☆47Updated 2 months ago
• webdataset / tarp
  Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.
  ☆132Updated last year
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆123Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated this week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆331Updated 2 years ago
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated 2 years ago
• google / qwix
  a Jax quantization library
  ☆56Updated this week
• lucidrains / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆17Updated 6 months ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆195Updated this week
• microsoft / ResiDual
  ResiDual: Transformer with Dual Residual Connections, https://arxiv.org/abs/2304.14802
  ☆96Updated 2 years ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆220Updated last year
• jax-ml / jax-tpu-embedding
  ☆24Updated last week
• riverstone496 / awesome-second-order-optimization
  ☆28Updated last month
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆253Updated 2 years ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• google / praxis
  ☆190Updated 2 weeks ago