jax-ml/australis external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

jax-ml/australis

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/jax-ml/australis)

Close

jax-ml / australisView on GitHubMore

• External links
• Readme badge

☆28Nov 18, 2022Updated 3 years ago

Alternatives and similar repositories for australis

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

• nestordemeure / jochastic

  View on GitHub
  A JAX implementation of stochastic addition.
  ☆14Aug 15, 2022Updated 3 years ago
• Sea-Snell / JAXSeq

  View on GitHub
  Train very large language models in Jax.
  ☆210Oct 21, 2023Updated 2 years ago
• AlirezaAzadbakht / kernel-sharing

  View on GitHub
  Drastically Reducing the Number of Trainable Parameters in Deep CNNs by Inter-layer Kernel-sharing
  ☆14Mar 28, 2023Updated 2 years ago
• deterministic-algorithms-lab / Jax-Journey

  View on GitHub
  A pathway and collection of resources to learning Jax from beginning to advance.
  ☆11Jan 2, 2021Updated 5 years ago
• catid / dataloader

  View on GitHub
  High-performance tokenized language data-loader for Python C++ extension
  ☆14Jul 22, 2024Updated last year
• google / timecast

  View on GitHub
  Performant, composable online learning
  ☆16Feb 22, 2021Updated 5 years ago
• kyunghyuncho / jax-practice

  View on GitHub
  ☆13Aug 17, 2020Updated 5 years ago
• jquesnelle / ctranslate2-rs

  View on GitHub
  Rust bindings for CTranslate2
  ☆14Jun 21, 2023Updated 2 years ago
• srush / learns-dex

  View on GitHub
  ☆33Jan 14, 2021Updated 5 years ago
• brendel-group / compositional-ood-generalization

  View on GitHub
  Official code for the paper "Compositional Generalization from First Principles" (NeurIPS 2023)
  ☆14Jul 25, 2023Updated 2 years ago
• n2cholas / progan-flax

  View on GitHub
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12May 24, 2021Updated 4 years ago
• shingarov / MachineArithmetic

  View on GitHub
  A mathematical foundation for Smalltalk-25
  ☆19Feb 21, 2026Updated last week
• GallagherCommaJack / modulax

  View on GitHub
  ☆18Aug 24, 2024Updated last year
• crazy4pi314 / pydataglobal2022

  View on GitHub
  PyData Global Workshop: Jupyter Notebooks in VS Code
  ☆15Dec 2, 2022Updated 3 years ago
• thecharlieblake / lovely-llama

  View on GitHub
  An implementation of the Llama architecture, to instruct and delight
  ☆21May 31, 2025Updated 9 months ago
• recursal / GoldFinch-paper

  View on GitHub
  GoldFinch and other hybrid transformer components
  ☆45Jul 20, 2024Updated last year
• aidangomez / welcome

  View on GitHub
  Generate a cute welcome message for yourself each day
  ☆22Mar 30, 2023Updated 2 years ago
• davisyoshida / haiku-mup

  View on GitHub
  A port of muP to JAX/Haiku
  ☆25Oct 23, 2022Updated 3 years ago
• umangjpatel / kerax

  View on GitHub
  Keras-like APIs for JAX framework
  ☆50Mar 25, 2023Updated 2 years ago
• gustavo-beck / wavebender-gan

  View on GitHub
  ☆24Sep 27, 2022Updated 3 years ago
• luohongyin / EntST

  View on GitHub
  Entailment self-training
  ☆27May 30, 2023Updated 2 years ago
• mhutchin-isolabs / ExtrinsicGaugeIndependentVectorGPs

  View on GitHub
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆26Oct 28, 2021Updated 4 years ago
• sholtodouglas / scalingExperiments

  View on GitHub
  ☆63Mar 4, 2022Updated 4 years ago
• MatX-inc / seqax

  View on GitHub
  seqax = sequence modeling + JAX
  ☆186Jul 23, 2025Updated 7 months ago
• expz / annotated-hyena

  View on GitHub
  An annotated implementation of the Hyena Hierarchy paper
  ☆34May 28, 2023Updated 2 years ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen

  View on GitHub
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆25Aug 16, 2023Updated 2 years ago
• brentyi / jax_dataclasses

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆75Dec 19, 2025Updated 2 months ago
• patrick-kidger / quax

  View on GitHub
  Multiple dispatch over abstract array types in JAX.
  ☆137Dec 15, 2025Updated 2 months ago
• google-research / precondition

  View on GitHub
  ☆31Jan 23, 2026Updated last month
• danieljtait / jax_xla_adventures

  View on GitHub
  Dive into Jax, Flax, XLA and C++
  ☆32Apr 1, 2020Updated 5 years ago
• google / init2winit

  View on GitHub
  ☆81Updated this week
• sayakpaul / ViT-jax2tf

  View on GitHub
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Mar 23, 2022Updated 3 years ago
• irhum / hyena

  View on GitHub
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Apr 27, 2023Updated 2 years ago
• crowsonkb / jax-wavelets

  View on GitHub
  The 2D discrete wavelet transform for JAX
  ☆45Feb 28, 2023Updated 3 years ago
• shawwn / ml-notes

  View on GitHub
  ☆40Dec 1, 2022Updated 3 years ago
• galilai-group / llm-jepa

  View on GitHub
  ☆182Feb 1, 2026Updated last month
• EOSIO / taurus-node

  View on GitHub
  EOSIO-Taurus - The Most Powerful Infrastructure for Decentralized Applications
  ☆13Mar 29, 2024Updated last year
• F1R3FLY-io / f1r3node

  View on GitHub
  ☆14Updated this week
• kinverarity1 / aseg_gdf2

  View on GitHub
  Python code to help read ASEG GDF2 packages
  ☆11Feb 17, 2026Updated 2 weeks ago