Alternatives and similar repositories for human_descent:

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

• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆81Updated last month
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 10 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆89Updated 8 months ago
• jax-ml / jax-llm-examples
  ☆102Updated this week
• Artur-Galstyan / statedict2pytree
  ☆43Updated last month
• modula-systems / modula
  🧱 Modula software package
  ☆188Updated 3 weeks ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆187Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆89Updated 3 weeks ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆136Updated 10 months ago
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated 8 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆111Updated 4 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆60Updated 8 months ago
• iliao2345 / CompressARC
  ☆143Updated 2 weeks ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆272Updated 10 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆62Updated 8 months ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆51Updated 5 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 9 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆80Updated 4 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆182Updated 3 weeks ago
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 3 weeks ago
• Future-House / aviary
  Gymnasium framework for training language model agents on constructive tasks
  ☆158Updated last week
• nikhilvyas / SOAP
  ☆175Updated 4 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• xjdr-alt / llmri
  look how they massacred my boy
  ☆63Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• riverstone496 / awesome-second-order-optimization
  ☆25Updated last year
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆94Updated 3 months ago
• shikaiqiu / compute-better-spent
  ☆52Updated 6 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 3 months ago