Alternatives and similar repositories for australis

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

• Sea-Snell / JAXSeq

  View on GitHub
  Train very large language models in Jax.
  ☆210Oct 21, 2023Updated 2 years ago
• catid / dataloader

  View on GitHub
  High-performance tokenized language data-loader for Python C++ extension
  ☆14Jul 22, 2024Updated last year
• 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
• 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)
  ☆13Jul 25, 2023Updated 2 years ago
• flyteorg / flyte-conference-talks

  View on GitHub
  A repo of Flyte-related conference talks
  ☆14Feb 24, 2024Updated last year
• thevasudevgupta / speech-jax

  View on GitHub
  Speech in Flax/JAX
  ☆15Jul 11, 2022Updated 3 years ago
• GallagherCommaJack / modulax

  View on GitHub
  ☆18Aug 24, 2024Updated last year
• shingarov / MachineArithmetic

  View on GitHub
  A mathematical foundation for Smalltalk-25
  ☆19Jan 30, 2026Updated 2 weeks ago
• thecharlieblake / lovely-llama

  View on GitHub
  An implementation of the Llama architecture, to instruct and delight
  ☆21May 31, 2025Updated 8 months ago
• jax-ml / jax-tpu-embedding

  View on GitHub
  ☆26Updated this week
• crazy4pi314 / pydataglobal2022

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

  View on GitHub
  A port of muP to JAX/Haiku
  ☆25Oct 23, 2022Updated 3 years ago
• gustavo-beck / wavebender-gan

  View on GitHub
  ☆24Sep 27, 2022Updated 3 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
• MatX-inc / seqax

  View on GitHub
  seqax = sequence modeling + JAX
  ☆170Jul 23, 2025Updated 6 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
• patrick-kidger / quax

  View on GitHub
  Multiple dispatch over abstract array types in JAX.
  ☆136Dec 15, 2025Updated last month
• danieljtait / jax_xla_adventures

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

  View on GitHub
  ☆31Jan 23, 2026Updated 3 weeks 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
• EOSIO / taurus-node

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

  View on GitHub
  Streamlit apps on Cloud Run with Identity-Aware Proxy (IAP).
  ☆10Mar 5, 2022Updated 3 years ago
• vvvm23 / TchAIkovsky

  View on GitHub
  Using JAX to generate piano music as MIDI
  ☆39Nov 28, 2023Updated 2 years ago
• Narsil / bloomserver

  View on GitHub
  ☆39Oct 3, 2022Updated 3 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite

  View on GitHub
  ☆36Nov 10, 2021Updated 4 years ago
• HomebrewML / TrueGrad

  View on GitHub
  PyTorch interface for TrueGrad Optimizers
  ☆43Aug 8, 2023Updated 2 years ago
• google-deepmind / distrax

  View on GitHub
  ☆622Jan 23, 2026Updated 3 weeks ago
• darribas / gds_ufmg19

  View on GitHub
  Geographic Data Science in Python - UFMG'19
  ☆12Mar 26, 2019Updated 6 years ago
• gdal4al / gdal-examples

  View on GitHub
  This is a repository for the Geospatial Data Abstraction Library (GDAL) and it's applications, examples and discussions in the world of s…
  ☆10May 28, 2023Updated 2 years ago
• simpeg / transform-2021-simpeg

  View on GitHub
  ☆14Apr 20, 2021Updated 4 years ago
• gailin-p / crust-composition-release

  View on GitHub
  Release code for "A Bayesian formulation for estimating the composition of Earth's crust"
  ☆10Apr 16, 2023Updated 2 years ago
• opendatacube / odc-stats

  View on GitHub
  Statistician is a framework of tools for generating statistical summaries of large collections of EO data managed in an ODC instance.
  ☆12Jan 27, 2026Updated 2 weeks ago
• ShuhongChen / vtubers-dataset

  View on GitHub
  CVPR 2023: PAniC-3D, Vtubers dataset downloader
  ☆13Apr 22, 2023Updated 2 years ago