jax

Metode Elemen Hingga Terdiferensiasi dengan JAX

Tevatron - Toolkit Pengambilan Dokumen Terpadu lintas Skala, Bahasa, dan Modalitas. Demo di SIGIR 2023, SIGIR 2025.

Model OCR yang ringan & cepat untuk pengenalan teks pelat nomor.

Panduan Machine Learning. Pelajari semua tentang Alat, Pustaka, Framework, Large Language Models (LLM), dan Pelatihan Model Machine Learning.

Algoritma pembelajaran mesin untuk sistem kuantum banyak-bodi

Tutorial Machine Learning Berbasis Fisika (Pytorch dan Jax)

Proses Gaussian di JAX dan Equinox.

Lingkungan game RL tervektorisasi di JAX

Paket Dinamika Fluida Terdiferensiasi

Optimisasi nonlinier (pencarian akar, kuadrat terkecil, ...) di JAX+Equinox. https://docs.kidger.site/optimistix/

Linear solver di JAX dan Equinox. https://docs.kidger.site/lineax

Menyelesaikan masalah maju dan mundur yang berkaitan dengan persamaan diferensial parsial menggunakan physics-informed neural networks berbasis basis berhingga (FBPINNs)

Semua yang ingin Anda ketahui tentang Google Cloud TPU

Felafax sedang membangun infrastruktur AI untuk GPU non-NVIDIA

Dari guru ke tile — mesin distilasi & penyajian LLM dari awal: kernel Triton/CUDA kustom, distilasi FSDP, batching kontinu paged-KV, dekoding spekulatif, gateway Rust, oracle JAX, dan alat interpretabilitas.

Pax adalah framework machine learning berbasis Jax untuk melatih model skala besar. Pax memungkinkan eksperimentasi dan paralelisasi tingkat lanjut yang sepenuhnya dapat dikonfigurasi, serta telah mendemonstrasikan tingkat pemanfaatan flop model yang terdepan di industri.

Komunikasi MPI zero-copy dari array JAX, untuk aplikasi HPC berkecepatan tinggi dengan Python :zap:

Orbax menyediakan utilitas checkpointing dan persistensi umum bagi pengguna JAX

Implementasi Keras dari model Yahoo Open-NSFW

Implementasi dari https://srush.github.io/annotated-s4

Model predictive controller untuk robot berkaki empat berdasarkan model bodi kaku tunggal yang ditulis dalam python. Berbasis gradien (acados) atau berbasis sampling (jax)

XLB: Accelerated Lattice Boltzmann (XLB) untuk ML berbasis fisika