A beginner-friendly guide to learning JAX with practical examples.
☆72May 18, 2026Updated 3 months ago
Alternatives and similar repositories for learn-jax
Users that are interested in learn-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆20Aug 6, 2025Updated last year
- ☆13May 27, 2025Updated last year
- AI-Powered pseudocode compiler 🚀☆18Jul 7, 2025Updated last year
- A fast, differentiable, and extensible public BBN code☆20Aug 31, 2026Updated last week
- An MLX port of Meta's Coconut reasoning model☆16Sep 2, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated last year
- Create one of the iconic figures in cosmology, from https://arxiv.org/abs/1905.08103 (thanks to Docker/Julia/Jupyter)☆20May 25, 2021Updated 5 years ago
- Hypernetwork training considerations and implementation types in PyTorch. Includes classification and time-series examples alongside 1D G…☆25Jan 4, 2023Updated 3 years ago
- Static file serving and directory listing☆12Apr 21, 2026Updated 4 months ago
- Make triton easier☆49Updated this week
- DISCO-DJ: DIfferentiable Simulations for COsmology - Done with Jax☆27May 16, 2026Updated 3 months ago
- Quadra: Effortless and reproducible deep learning workflows with configuration files.☆51May 21, 2026Updated 3 months ago
- Code to accompany the paper "Fast Hamiltonian Monte Carlo Using GPU Computing"☆37Nov 3, 2016Updated 9 years ago
- Use local LLMs for advanced NLP☆38Mar 19, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- JAX-based linear Einstein-Boltzmann solver for cosmology☆27Nov 13, 2025Updated 9 months ago
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- This is a proof of concept that shows how you can get cursor to execute tasks like creating images. It require the new composer multi fil…☆13Aug 6, 2024Updated 2 years ago
- Skill file for MLX-Swift-LM☆24Feb 24, 2026Updated 6 months ago
- Notes of ADRL course taught at IISC as part of MTech AI curriculum☆16Nov 30, 2024Updated last year
- GPT2 in handwritten PTX☆16Jun 29, 2025Updated last year
- A Swift wrapper for the Supertone text-to-speech model☆34Dec 11, 2025Updated 8 months ago
- ☆15Feb 23, 2025Updated last year
- Welcome to The C Programming Language. This repository is a collection of my journey through the world of C programming. Here, you'll fin…☆13Oct 7, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Fetch arxiv data to LLM-friendly text☆133Feb 18, 2026Updated 6 months ago
- This repo maintains a 'cheat sheet' for LLMs that are undertrained on mlx☆33Mar 12, 2026Updated 5 months ago
- vibevoice real time 0.5B swift port☆31Dec 12, 2025Updated 8 months ago
- ☆22Mar 20, 2024Updated 2 years ago
- Terraform Script for - Storage, container and data life cycle rules creation at scale☆11Jan 10, 2023Updated 3 years ago
- A fork of adoptium/aqa-tests with Msft specific changes☆13Jun 29, 2026Updated 2 months ago
- This repository stores the files used for my summer internship's work on "teacher-student learning", an experimental method for training …☆48Jan 26, 2019Updated 7 years ago
- ☆20Aug 23, 2022Updated 4 years ago
- A simple flux tutorial built in typescript with alt and react☆12Jun 30, 2015Updated 11 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Training Neural Network potentials through customizable routines in JAX.☆73Updated this week
- Terraform Azure Verified Resource Module for Container App☆14Updated this week
- Improving langchain knowledge graphs using baml☆46Aug 3, 2025Updated last year
- Code to check if there is a new version on the App Store.☆20Apr 23, 2026Updated 4 months ago
- Easily manage AWS CDK Dependencies☆15Mar 27, 2022Updated 4 years ago
- A simple way to learn about GitHub☆10Aug 9, 2024Updated 2 years ago
- Notes of PRNN course taught at IISC as part of MTech AI curriculum☆21Nov 30, 2024Updated last year