jax-ml / australis

Related projects: ⓘ

• google-research / precondition
  ☆27Updated 4 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆28Updated last year
• cgarciae / einop
  ☆56Updated 2 years ago
• andyehrenberg / flaxlm
  ☆27Updated this week
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆41Updated 3 months ago
• srush / Tensor-Puzzles-Penzai
  ☆17Updated 4 months ago
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆12Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆66Updated 7 months ago
• sholtodouglas / scalingExperiments
  ☆56Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆29Updated 2 months ago
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 8 months ago
• jax-ml / jax-ai-stack
  ☆28Updated this week
• shawwn / ml-notes
  ☆38Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆82Updated 2 months ago
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 3 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆29Updated last year
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆57Updated last month
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆36Updated last year
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated last year
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆57Updated 10 months ago
• cgarciae / jax-gravity
  ☆24Updated last year
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆31Updated last year
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆43Updated 3 weeks ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆66Updated last year
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated 11 months ago
• mgrankin / minGPT
  minGPT in JAX
  ☆45Updated 2 years ago
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated last year
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated last year