JAX models for ML
☆12Mar 2, 2026Updated 7 months ago
Alternatives and similar repositories for jaxonmodels
Users that are interested in jaxonmodels are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 9 months ago
- Heap Implementation for Python☆10Nov 4, 2016Updated 9 years ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- This repository shows how to efficiently process variable-length sequences in TensorFlow.☆14Apr 26, 2022Updated 4 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 11 months ago
- ☆13Nov 18, 2023Updated 2 years ago
- This repository contains code for evaluating the predicted uncertainty of second-order distributions, thereby comparing a "reference" res…☆11Jul 16, 2024Updated 2 years ago
- A short digital signal processing primer with example code snippets in Python☆12Jun 19, 2025Updated last year
- ☆14May 27, 2023Updated 3 years ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- ☆12Jul 2, 2024Updated 2 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- A generic flux/parameter population synthesis code☆14Sep 16, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Hierarchical Dirichlet Process (with Split-Merge Operations), originally by Chong Wang☆18Oct 12, 2013Updated 13 years ago
- Implementation of Deep Learning Models in JAX (equinox)☆25Oct 21, 2026Updated last week
- Jax implementation of a flexible audio loudness meter in Python with implementation of ITU-R BS.1770-4 loudness algorithm☆13Jan 29, 2025Updated last year
- ☆16Oct 31, 2025Updated 11 months ago
- jax port of snac☆11May 12, 2024Updated 2 years ago
- mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.☆19Jun 23, 2026Updated 4 months ago
- Xtructure is datastructure for using in JAX☆23Updated this week
- ☆15Sep 24, 2022Updated 4 years ago
- Official repo for Trumpets: Injective Flows for Inference and Inverse Problems☆15Jun 2, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- High quality implementations of imitation and inverse reinforcement learning algorithms☆26Aug 19, 2025Updated last year
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- X-ray absorption re-emission scattering - Monte Carlo simulator for X-ray obscurers☆22Oct 13, 2026Updated 2 weeks ago
- Knowledgebase— a collection of information for quantitative finance, insurance, mathematics and AI—This serves as a sprawling notebook of…☆18Jun 4, 2025Updated last year
- ☆127Jun 11, 2025Updated last year
- ☆246Updated this week
- qwen3 experiments☆33Jul 1, 2025Updated last year
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- LearningByDoing NeurIPS 2021 Competition: Standalone Code and Results☆20Aug 30, 2022Updated 4 years ago
- VoxEMG circuit and schematics, plus example code for further digital filtering with Bela, etc.☆12Mar 6, 2024Updated 2 years ago
- Experimental text-to-video generation☆16Aug 30, 2026Updated last month
- A Variational Autoencoder in PyTorch for the CelebA Dataset.☆20Nov 1, 2023Updated 2 years ago
- BAD-VAE: A VAE framework for unsupervised disentanglement of sequential data☆12May 25, 2022Updated 4 years ago
- NeurIPS 2019: DQN(λ) = Deep Q-Network + λ-returns.☆25May 20, 2024Updated 2 years ago
- A screaming vocal samples dataset.☆13Apr 14, 2023Updated 3 years ago