jax

Método de Elementos Finitos Diferenciável com JAX

Tevatron - Kit de ferramentas unificado de recuperação de documentos em escala, idioma e modalidade. Demonstração no SIGIR 2023, SIGIR 2025.

Modelos de OCR leves e rápidos para reconhecimento de texto em placas de veículos.

Guia de aprendizado de máquina. Aprenda tudo sobre ferramentas, bibliotecas, frameworks, grandes modelos de linguagem (LLMs) e treinamento de modelos.

Algoritmos de aprendizado de máquina para sistemas quânticos de muitos corpos

Tutoriais de Aprendizado de Máquina Informados por Física (PyTorch e Jax)

Processos gaussianos em JAX e Equinox.

♟️ Ambientes de jogos de RL vetorizados em JAX

Pacote de dinâmica de fluidos diferenciável

Otimização não linear (busca de raízes, mínimos quadrados, ...) em JAX+Equinox. https://docs.kidger.site/optimistix/

Solutores lineares em JAX e Equinox. https://docs.kidger.site/lineax

Resolva problemas diretos e inversos relacionados a equações diferenciais parciais usando redes neurais informadas pela física de base finita (FBPINNs).

Tudo o que você quer saber sobre o Google Cloud TPU

Felafax está construindo infraestrutura de IA para GPUs não-NVIDIA

From teacher to tiles — a from-scratch LLM distillation & serving engine: custom Triton/CUDA kernels, FSDP distillation, paged-KV continuous batching, speculative decoding, a Rust gateway, a JAX oracle, and interpretability tooling.

Pax é um framework de aprendizado de máquina baseado em Jax para treinamento de modelos de grande escala. O Pax permite experimentação e paralelização avançadas e totalmente configuráveis, demonstrando taxas de utilização de FLOPs de modelo líderes do setor.

Comunicação MPI zero-copy de arrays JAX, para aplicações de HPC superaceleradas em Python :zap:

Orbax fornece utilitários comuns de checkpoint e persistência para usuários de JAX.

Implementação em Keras do modelo Yahoo Open-NSFW

Implementação de https://srush.github.io/annotated-s4

Um controlador preditivo baseado em modelo para robôs quadrúpedes, baseado no modelo de corpo rígido único e escrito em Python. Baseado em gradiente (acados) ou baseado em amostragem (jax)

XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML